Hugging Face's logo

Spaces:
E-Hospital
/
oop-deploy
Paused

App Files Community
E-Hospital commited on
Commit
3de3aa5
•
1 Parent(s): 81215d7

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -7,8 +7,7 @@ device = torch.device("cuda")
7
 
8
  model = AutoModelForCausalLM.from_pretrained(
9
  "E-Hospital/open-orca-platypus-2-lora-medical",
10
- trust_remote_code=True,
11
- device_map = "auto",
12
  )
13
  tokenizer = AutoTokenizer.from_pretrained("Open-Orca/OpenOrca-Platypus2-13B", trust_remote_code=True)
14
 
 
7
 
8
  model = AutoModelForCausalLM.from_pretrained(
9
  "E-Hospital/open-orca-platypus-2-lora-medical",
10
+ trust_remote_code=True
 
11
  )
12
  tokenizer = AutoTokenizer.from_pretrained("Open-Orca/OpenOrca-Platypus2-13B", trust_remote_code=True)
13