Salesforce
/

codet5p-6b

Text2Text Generation

Model card Files Files and versions Community

yuewang-sf commited on May 17, 2023

Commit

70e0c1e

•

1 Parent(s): 0d1bc98

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -31,8 +31,8 @@ model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint,
                                               torch_dtype=torch.float16,
                                               trust_remote_code=True).to(device)
-inputs = tokenizer.encode("def print_hello():", return_tensors="pt").to(device)
-outputs = model.generate(inputs, max_length=12)
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 ```

                                               torch_dtype=torch.float16,
                                               trust_remote_code=True).to(device)
+inputs = tokenizer.encode("def print_hello_world():", return_tensors="pt").to(device)
+outputs = model.generate(inputs, max_length=8)
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 ```