From 4a770ac22b85e70de5bafd5f6b68118dddcef2fd Mon Sep 17 00:00:00 2001
From: City <125218114+city96@users.noreply.github.com>
Date: Fri, 13 Dec 2024 17:46:53 +0100
Subject: [PATCH] Add the non-gated repo

---
 Gemma/nodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Gemma/nodes.py b/Gemma/nodes.py
index 9781ea1..3446f40 100644
--- a/Gemma/nodes.py
+++ b/Gemma/nodes.py
@@ -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,),
             }