jphme jncraton commited on
Commit
04dc770
1 Parent(s): ccad403

Correct undefined variable in example (#1)

Browse files

- Correct undefined variable in example (547e4b1cbe0b0b9eb70a573e8e7d0ba325b049fd)


Co-authored-by: Jon <jncraton@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +1 -1
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 = model_wiz.generate(
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,