Spaces:

as-cle-bert
/

bloom-multilingual-chat

Sleeping

as-cle-bert commited on May 16

Commit

7f74d90

•

1 Parent(s): bde06a8

Update load_model.py

Files changed (1) hide show

load_model.py CHANGED Viewed

@@ -1,6 +1,6 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
-model = AutoModelForCausalLM.from_pretrained("h2oai/h2o-danube2-1.8b-chat")
-tokenizer = AutoTokenizer.from_pretrained("h2oai/h2o-danube2-1.8b-chat")
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=1024, repetition_penalty=1.2, temperature=0.4)

 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+model = AutoModelForCausalLM.from_pretrained("EleutherAI/pythia-2.8b")
+tokenizer = AutoTokenizer.from_pretrained("EleutherAI/pythia-2.8b")
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=1024, repetition_penalty=1.2, temperature=0.4)