Spaces:

LLM-DEMO
/

llama-2-medical-consultation

Paused

Ashishkr commited on Aug 30, 2023

Commit

02cddd1

•

1 Parent(s): 3697a24

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -28,7 +28,7 @@ model = transformers.AutoModelForCausalLM.from_pretrained(
     config=model_config,
     quantization_config=bnb_config,
     device_map='auto',
-    use_auth_token=hf_auth
 )
 config = PeftConfig.from_pretrained("Ashishkr/llama-2-medical-consultation")

     config=model_config,
     quantization_config=bnb_config,
     device_map='auto',
+    use_auth_token=token
 )
 config = PeftConfig.from_pretrained("Ashishkr/llama-2-medical-consultation")