Update nodes.py

2024-12-13 19:02:26 +01:00
parent 4a770ac22b
commit 6e5ad55f40
1 changed files with 4 additions and 0 deletions
@@ -56,6 +56,10 @@ class GemmaLoader:
            text_encoder_dir = os.path.join(folder_paths.models_dir, 'text_encoders', 'models--unsloth--gemma-2-2b-it-bnb-4bit')
            if not os.path.exists(os.path.join(text_encoder_dir, 'model.safetensors')):
                snapshot_download('unsloth/gemma-2-2b-it-bnb-4bit', local_dir=text_encoder_dir)
+        elif model_name == 'Efficient-Large-Model/gemma-2-2b-it':
+            text_encoder_dir = os.path.join(folder_paths.models_dir, 'text_encoders', 'models--Efficient-Large-Model--gemma-2-2b-it')
+            if not os.path.exists(os.path.join(text_encoder_dir, 'model.safetensors')):
+                snapshot_download('Efficient-Large-Model/gemma-2-2b-it', local_dir=text_encoder_dir)
        else:
            raise ValueError('Not implemented!')