Spaces:

JeffMII
/

CEC-Learning

Running

Jeff Myers II commited on Apr 21

Commit

1073f17

1 Parent(s): eb00346

Update space

Files changed (1) hide show

Gemma_Model.py CHANGED Viewed

@@ -21,7 +21,7 @@ class GemmaLLM:
             # quantization_config=quantization_config,
             # low_cpu_mem_usage=True,
             # torch_dtype=torch.float16,
-            # token=os.environ.get("GEMMA_TOKEN"),
         ).eval()
         self.tokenizer = AutoTokenizer.from_pretrained(model_id)

             # quantization_config=quantization_config,
             # low_cpu_mem_usage=True,
             # torch_dtype=torch.float16,
+            token=os.environ.get("GEMMA_TOKEN"),
         ).eval()
         self.tokenizer = AutoTokenizer.from_pretrained(model_id)