iamtarun
/

pycompetitive-codegen350M-qlora

Text Generation

Model card Files Files and versions

iamtarun commited on Jul 28, 2023

Commit

f5ea8d4

·

1 Parent(s): 43ab4b7

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -56,8 +56,8 @@ def generate_prompt(description, inputs, outputs):
 from transformers import AutoModelForCausalLM, AutoTokenizer
 # load model and tokenizer
-model = AutoModelForCausalLM.from_pretrained(args.model_path, device_map="auto")
-tokenizer = AutoTokenizer.from_pretrained(args.model_path)
 # loading model for inference
 model.eval()

 from transformers import AutoModelForCausalLM, AutoTokenizer
 # load model and tokenizer
+model = AutoModelForCausalLM.from_pretrained("iamtarun/pycompetitive-codegen350M-qlora", device_map="auto")
+tokenizer = AutoTokenizer.from_pretrained("iamtarun/pycompetitive-codegen350M-qlora")
 # loading model for inference
 model.eval()