ssong1
/

gpt-j-5.8b-sum-adapter

Model card Files Files and versions Community

ssong1 commited on Jan 16

Commit

0ce4c7d

•

1 Parent(s): 07fdc61

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -24,10 +24,10 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 from peft import PeftModel
 model1 = AutoModelForCausalLM.from_pretrained(
-    "ssong1/gpt-j-5.8b-sum_adapter", torch_dtype="auto", device_map="auto"
 )
-lora_path = "ssong1/dssong1/gpt-j-5.8b-sum-adapter"
 model2 = PeftModel.from_pretrained(model1, lora_path, device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained(lora_path)

 from peft import PeftModel
 model1 = AutoModelForCausalLM.from_pretrained(
+    "ssong1/gpt-j-5.8b", torch_dtype="auto", device_map="auto"
 )
+lora_path = "ssong1/gpt-j-5.8b-sum-adapter"
 model2 = PeftModel.from_pretrained(model1, lora_path, device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained(lora_path)