Release 1.3.0

iejMac · Oct 11, 2022 · 00bc173 · 00bc173
1 parent aef59e3
commit 00bc173
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/HISTORY.md b/HISTORY.md
@@ -1,3 +1,9 @@
+## 1.3.0
+
+* Transition from OpenAI CLIP to open_clip CLIP + option to select between available CLIP models
+* Bug fix - correct video-text alignment for large dataset creation tasks
+* Distributing jobs across multiple Slurm nodes is now possible
+
 ## 1.2.0
 
 * Take parquet or csv as input and write metadata and captions

diff --git a/README.md b/README.md
@@ -49,6 +49,10 @@ DESCRIPTION
         bool: use the save name suggested by video2numpy
       distribute:
         str: distribution strategy, currently either slurm or none
+      oc_model_name:
+        str: open_clip model name, used for selecting CLIP architecture
+      pretrained:
+        str: open_clip pretrained weights name
 
 POSITIONAL ARGUMENTS
     SRC
@@ -70,6 +74,10 @@ FLAGS
         Default: False
     --distribute=DISTRIBUTE
         Default: 'none'
+    --oc_model_name=OC_MODEL_NAME
+        Default: 'ViT-B-32'
+    --pretrained=PRETRAINED
+        Default: 'laion2b_s34b_b79k'
 ```
 
 ## API

diff --git a/setup.py b/setup.py
@@ -17,7 +17,7 @@ def _read_reqs(relpath):
         name="clip-video-encode",
         packages=find_packages(),
         include_package_data=True,
-        version="1.2.0",
+        version="1.3.0",
         license="MIT",
         description="Easily compute clip embeddings from video frames",
         long_description=long_description,