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

405bbd1

·

1 Parent(s): b1db2e4

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -62,7 +62,7 @@ model.eval()
 print(f"Model loaded on {device}")
 tokenizer = transformers.AutoTokenizer.from_pretrained(
-    model_id,
     use_auth_token=hf_auth
 )

 print(f"Model loaded on {device}")
 tokenizer = transformers.AutoTokenizer.from_pretrained(
+    base_model_id,
     use_auth_token=hf_auth
 )