added ViT-L/14_336px support

src/pytti/Notebook.py

@@ -215,6 +215,8 @@ def save_batch(settings_list, path):
 def _sanitize_for_config(in_str):
     for char in ("/", "-"):
         in_str = in_str.replace(char, "")
+    for char in "@":
+        in_str = in_str.replace(char, "_")
     return in_str
 
 

src/pytti/assets/default.yaml

@@ -162,6 +162,7 @@ flow_long_term_samples: 1
 ViTB32: true
 ViTB16: false
 ViTL14: false
+ViTL14_336px: false
 RN50: false
 RN101: false
 RN50x4: false

tests/test_models_load.py

@@ -63,28 +63,32 @@ def test_init_vqgan(self, model_name):
 
 ########################################
 
+# clip
 
-def test_load_vqgan_with_download():
-    pass
+from pytti.Notebook import load_clip, _sanitize_for_config
 
 
-# clip
+def test_load_clip_w_at_symbol_in_identifier():
+    openai_model_name = "ViT-L/14@336px"
+    config_model_name = "ViTL14_336px"
+    assert _sanitize_for_config(openai_model_name) == config_model_name
+    load_clip({config_model_name: True})
 
 
-def test_load_clip_with_download():
-    pass
+# def test_load_clip_with_download():
+#    pass
 
 
-def test_load_clip_no_download():
-    pass
+# def test_load_clip_no_download():
+#    pass
 
 
 # adabins
 
 
-def test_load_adabins_with_download():
-    pass
+# def test_load_adabins_with_download():
+#    pass
 
 
-def test_load_adabins_no_download():
-    pass
+# def test_load_adabins_no_download():
+#    pass