Spaces:

acbdkk
/

gptj

Runtime error

acbdkk commited on Oct 14, 2023

Commit

0c7c9e7

•

1 Parent(s): 5429516

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ model = "PY007/TinyLlama-1.1B-Chat-v0.1"
 tokenizer = transformers.LlamaTokenizer.from_pretrained(model)
 pipeline = transformers.pipeline("text-generation",model=model,torch_dtype=torch.bfloat16)
-prompt = "What is 6534+2343?"
 formatted_prompt = (
     f"### Human: {prompt}### Assistant:"
 )

 tokenizer = transformers.LlamaTokenizer.from_pretrained(model)
 pipeline = transformers.pipeline("text-generation",model=model,torch_dtype=torch.bfloat16)
+prompt = "What is 6534+2343?"
 formatted_prompt = (
     f"### Human: {prompt}### Assistant:"
 )