deepseek-ai
/

deepseek-coder-33b-base

Text Generation

text-generation-inference

Model card Files Files and versions

zqh11 commited on Nov 3, 2023

Commit

9a4db57

·

1 Parent(s): fd806a6

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -144,7 +144,7 @@ from model import IrisClassifier as Classifier
 def main():
     # Model training and evaluation
 """
-inputs = tokenizer(input_text, return_tensors="pt").cuda()
 outputs = model.generate(**inputs, max_new_tokens=140)
 print(tokenizer.decode(outputs[0]))
 ```

 def main():
     # Model training and evaluation
 """
+inputs = tokenizer(input_text, return_tensors="pt").to(model.device)
 outputs = model.generate(**inputs, max_new_tokens=140)
 print(tokenizer.decode(outputs[0]))
 ```