VCoder / Run
varuneshv's picture
new run info
42c1a58 verified
Raw
History Blame Contribute Delete
507 Bytes
Step1:
!pip install -U transformers
step2:
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("varuneshv/VCoder")
model = AutoModelForCausalLM.from_pretrained(
"varuneshv/VCoder"
)
step3:
model = model.to("cuda")
inputs = tokenizer(
"write a python code to merge 3 arrays",
return_tensors="pt"
).to("cuda")
outputs = model.generate(
**inputs,
max_new_tokens=100
)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))