PipableAI
/

pip-code-bandit

Model card Files Files and versions Community

QagentS commited on May 13, 2024

Commit

4f8f251

·

verified ·

1 Parent(s): 9fbe2b8

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -116,7 +116,7 @@ model =AutoModelForCausalLM.from_pretrained("PipableAI/pip-code-bandit",torch_dt
 tokenizer = tokenizer = AutoTokenizer.from_pretrained("PipableAI/pip-code-bandit")
 inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
-outputs = model.generate(**inputs, max_new_tokens=new_tokens,)
 out = (
     tokenizer.decode(outputs[0][inputs.input_ids.shape[-1]:], skip_special_tokens=True)
 )

 tokenizer = tokenizer = AutoTokenizer.from_pretrained("PipableAI/pip-code-bandit")
 inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
+outputs = model.generate(**inputs, max_new_tokens=new_tokens)
 out = (
     tokenizer.decode(outputs[0][inputs.input_ids.shape[-1]:], skip_special_tokens=True)
 )