E-Hospital commited on
Commit
e26afaa
1 Parent(s): e9875b6

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -6,7 +6,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
6
 
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)
 
6
 
7
 
8
  model = AutoModelForCausalLM.from_pretrained(
9
+ "E-Hospital/open-orca-platypus-2-lora-latest",
10
  trust_remote_code=True
11
  )
12
  tokenizer = AutoTokenizer.from_pretrained("Open-Orca/OpenOrca-Platypus2-13B", trust_remote_code=True)