Spaces:

hereoncollab
/

llama

Runtime error

hereoncollab commited on Sep 17

Commit

10ceff5

•

1 Parent(s): 0d5e624

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ enr_token = MarianTokenizer.from_pretrained(enr_md)
 enr_model = MarianMTModel.from_pretrained(enr_md)
 # Load the Gemma model for text generation, ensuring it runs on CPU
-gemma_model = "google/gemma-2-2b-it"
 gemma_tokenizer = AutoTokenizer.from_pretrained(gemma_model)
 pipe = pipeline(

 enr_model = MarianMTModel.from_pretrained(enr_md)
 # Load the Gemma model for text generation, ensuring it runs on CPU
+gemma_model = "stabilityai/stablelm-2-1_6b-chat"
 gemma_tokenizer = AutoTokenizer.from_pretrained(gemma_model)
 pipe = pipeline(