Spaces:

as-cle-bert
/

bloom-multilingual-chat

Sleeping

as-cle-bert commited on May 16

Commit

2c4e93b

•

1 Parent(s): 7f74d90

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("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)

 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen1.5-1.8B-Chat")
+tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-1.8B-Chat")
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=1024, repetition_penalty=1.2, temperature=0.4)