protgpt3 commited on
Commit
778ed06
·
verified ·
1 Parent(s): 6d7bb4e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -106,7 +106,7 @@ inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
106
 
107
  with torch.no_grad():
108
  output_ids = model.generate(
109
- **inputs,
110
  max_new_tokens=512,
111
  do_sample=True,
112
  temperature=0.8,
@@ -131,7 +131,7 @@ inputs = tokenizer(prefix, return_tensors="pt").to(model.device)
131
 
132
  with torch.no_grad():
133
  output_ids = model.generate(
134
- **inputs,
135
  max_new_tokens=256,
136
  do_sample=True,
137
  temperature=0.8,
@@ -163,7 +163,7 @@ inputs = tokenizer(
163
 
164
  with torch.no_grad():
165
  output_ids = model.generate(
166
- **inputs,
167
  max_new_tokens=256,
168
  do_sample=True,
169
  temperature=0.8,
 
106
 
107
  with torch.no_grad():
108
  output_ids = model.generate(
109
+ inputs["input_ids"],
110
  max_new_tokens=512,
111
  do_sample=True,
112
  temperature=0.8,
 
131
 
132
  with torch.no_grad():
133
  output_ids = model.generate(
134
+ inputs["input_ids"],
135
  max_new_tokens=256,
136
  do_sample=True,
137
  temperature=0.8,
 
163
 
164
  with torch.no_grad():
165
  output_ids = model.generate(
166
+ inputs["input_ids"],
167
  max_new_tokens=256,
168
  do_sample=True,
169
  temperature=0.8,