Ashishkr
/

llama-2-medical-consultation

Text Generation

Trained with AutoTrain

meta-llama/Llama-2-7b-hf

Model card Files Files and versions Metrics Training metrics Community

Ashishkr commited on Aug 30, 2023

Commit

a85c56a

•

1 Parent(s): 405bbd1

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -55,8 +55,8 @@ model = transformers.AutoModelForCausalLM.from_pretrained(
     use_auth_token=hf_auth
 )
-config = PeftConfig.from_pretrained("Ashishkr/llama2_medical_consultation")
-model = PeftModel.from_pretrained(model, "Ashishkr/llama2_medical_consultation").to(device)
 model.eval()
 print(f"Model loaded on {device}")

     use_auth_token=hf_auth
 )
+config = PeftConfig.from_pretrained("Ashishkr/llama-2-medical-consultation")
+model = PeftModel.from_pretrained(model, "Ashishkr/llama-2-medical-consultation").to(device)
 model.eval()
 print(f"Model loaded on {device}")