prithivida
/

Asimov-7B-v1

Text Generation

Model card Files Files and versions

prithivida commited on Nov 18, 2023

Commit

0b33ad0

·

1 Parent(s): 38cf961

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -56,7 +56,7 @@ base_model = AutoModelForCausalLM.from_pretrained(
 model = PeftModel.from_pretrained(
         base_model,
-        './lora_adapter',
         torch_dtype=torch.float16,
         device_map="auto",
 )

 model = PeftModel.from_pretrained(
         base_model,
+        model_name,
         torch_dtype=torch.float16,
         device_map="auto",
 )