Add the non-gated repo

2024-12-13 17:46:53 +01:00
parent aec37b0ce7
commit 4a770ac22b
1 changed files with 1 additions and 1 deletions
@@ -33,7 +33,7 @@ class GemmaLoader:
            devices.append(f"cuda:{k}")
        return {
            "required": {
-                "model_name": (["google/gemma-2-2b-it", "unsloth/gemma-2-2b-it-bnb-4bit"],),
+                "model_name": (["Efficient-Large-Model/gemma-2-2b-it", "google/gemma-2-2b-it", "unsloth/gemma-2-2b-it-bnb-4bit"],),
                "device": (devices, {"default":"cpu"}),
                "dtype": (dtypes,),
            }