Correct undefined variable in example
Browse files`model_wiz` was undefined and presumably should simply be `model`.
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,
|