Spaces:

JeffMII
/

CEC-Learning

Running

Jeff Myers II commited on Apr 21

Commit

0b67bbe

1 Parent(s): 1073f17

Update space

Files changed (1) hide show

Gemma_Model.py CHANGED Viewed

@@ -24,7 +24,7 @@ class GemmaLLM:
             token=os.environ.get("GEMMA_TOKEN"),
         ).eval()
-        self.tokenizer = AutoTokenizer.from_pretrained(model_id)
     def generate(self, message) -> str:
         print("Generating...")

             token=os.environ.get("GEMMA_TOKEN"),
         ).eval()
+        self.tokenizer = AutoTokenizer.from_pretrained(model_id, force_download=True)
     def generate(self, message) -> str:
         print("Generating...")