release prep

ssundaram21 · Aug 18, 2024 · c70f64a · c70f64a
1 parent becf40b
commit c70f64a
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 17 deletions.
diff --git a/dreamsim/config.py b/dreamsim/config.py
@@ -28,6 +28,7 @@
         "img_size": 224
     }
 
+# UPDATE
 dreamsim_weights = {
     "ensemble": "https://github.com/ssundaram21/dreamsim/releases/download/v0.1.0/dreamsim_checkpoint.zip",
     "dino_vitb16": "https://github.com/ssundaram21/dreamsim/releases/download/v0.1.2/dreamsim_dino_vitb16_checkpoint.zip",

diff --git a/dreamsim/feature_extraction/load_synclr_as_dino.py b/dreamsim/feature_extraction/load_synclr_as_dino.py
diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
 
 setuptools.setup(
     name="dreamsim",
-    version="0.1.3",
+    version="0.2.0",
     description="DreamSim similarity metric",
     long_description=long_description,
     long_description_content_type="text/markdown",

diff --git a/training/download_models.sh b/training/download_models.sh
@@ -2,6 +2,7 @@
 mkdir -p ./models
 cd models
 
+## UDPATE
 wget -O dreamsim_checkpoint.zip https://github.com/ssundaram21/dreamsim/releases/download/v0.1.0/dreamsim_checkpoint.zip
 wget -O clip_vitb32_pretrain.pth.tar https://github.com/ssundaram21/dreamsim/releases/download/v0.1.0/clip_vitb32_pretrain.pth.tar
 wget -O clipl14_as_dino_vitl.pth.tar https://github.com/ssundaram21/dreamsim/releases/download/v0.1.0/clip_vitl14_pretrain.pth.tar