Correct undefined variable in example (#1)
Browse files- Correct undefined variable in example (547e4b1cbe0b0b9eb70a573e8e7d0ba325b049fd)
Co-authored-by: Jon <jncraton@users.noreply.huggingface.co>
README.md
CHANGED
@@ -43,7 +43,7 @@ prompt_template=("A chat between a curious user and an artificial intelligence a
|
|
43 |
inputs = tokenizer(
|
44 |
prompt_template.format(prompt="What is 1+1?"),
|
45 |
return_tensors="pt", return_attention_mask=False).to('cuda')
|
46 |
-
outputs =
|
47 |
**inputs, max_length=200,
|
48 |
do_sample=True,
|
49 |
temperature=0.5,
|
|
|
43 |
inputs = tokenizer(
|
44 |
prompt_template.format(prompt="What is 1+1?"),
|
45 |
return_tensors="pt", return_attention_mask=False).to('cuda')
|
46 |
+
outputs = model.generate(
|
47 |
**inputs, max_length=200,
|
48 |
do_sample=True,
|
49 |
temperature=0.5,
|