Spaces:

kuangren
/

dolly_test_1

Runtime error

kuangren commited on Apr 19, 2023

Commit

3d6af4a

•

1 Parent(s): c57d64a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ from PIL import Image
 from instruct_pipeline import InstructionTextGenerationPipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
-tokenizer = AutoTokenizer.from_pretrained("databricks/dolly-v2-12b", padding_side="left")
-model = AutoModelForCausalLM.from_pretrained("databricks/dolly-v2-12b", device_map="auto")
 generate_text = InstructionTextGenerationPipeline(model=model, tokenizer=tokenizer)
 text=generate_text("Explain to me the difference between nuclear fission and fusion.")

 from instruct_pipeline import InstructionTextGenerationPipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained("databricks/dolly-v2-3b", padding_side="left")
+model = AutoModelForCausalLM.from_pretrained("databricks/dolly-v2-3b", device_map="auto")
 generate_text = InstructionTextGenerationPipeline(model=model, tokenizer=tokenizer)
 text=generate_text("Explain to me the difference between nuclear fission and fusion.")