Spaces:

JeffMII
/

CEC-Learning

Running

Jeff Myers II commited on May 6

Commit

c500843

1 Parent(s): 7137977

Completed Prototype

Files changed (1) hide show

Gemma.py CHANGED Viewed

@@ -13,7 +13,7 @@ class GemmaLLM:
     def __init__(self):
         login(token=os.environ.get("GEMMA_TOKEN"))
-        model_id = "google/gemma-3-1b-it"
         # self.tokenizer = AutoTokenizer.from_pretrained(model_id)
         # self.model = Gemma3ForCausalLM.from_pretrained(

     def __init__(self):
         login(token=os.environ.get("GEMMA_TOKEN"))
+        model_id = "google/gemma-3-4b-it"
         # self.tokenizer = AutoTokenizer.from_pretrained(model_id)
         # self.model = Gemma3ForCausalLM.from_pretrained(