nodes/city96-ComfyUI_ExtraModels
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
City
2024-12-13 19:02:26 +01:00
committed by GitHub
parent 4a770ac22b
commit 6e5ad55f40
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Gemma/nodes.py
+4
View File
@@ -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!')