Spaces:

mohanchinnappan
/

sf-codegen

Runtime error

Create app.py

472d6f4 over 3 years ago

426 Bytes

from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen-16B-multi")
model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen-16B-multi")

text = "def hello_world():"
input_ids = tokenizer(text, return_tensors="pt").input_ids

generated_ids = model.generate(input_ids, max_length=128)
print(tokenizer.decode(generated_ids[0], skip_special_tokens=True))

	from transformers import AutoTokenizer, AutoModelForCausalLM
	tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen-16B-multi")
	model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen-16B-multi")

	text = "def hello_world():"
	input_ids = tokenizer(text, return_tensors="pt").input_ids

	generated_ids = model.generate(input_ids, max_length=128)
	print(tokenizer.decode(generated_ids[0], skip_special_tokens=True))