agarkovv commited on
Commit
72d9e7a
·
verified ·
1 Parent(s): 758f7f6

Update run_inference.py

Browse files
Files changed (1) hide show
  1. run_inference.py +1 -1
run_inference.py CHANGED
@@ -8,7 +8,7 @@ MAX_LENGTH = 32768 # Do not change
8
  DEVICE = "cuda"
9
 
10
 
11
- model_id = "agarkovv/Ministral-8B-Instruct-2410-LoRA-trading"
12
  base_model_id = "mistralai/Ministral-8B-Instruct-2410"
13
 
14
  model = AutoPeftModelForCausalLM.from_pretrained(model_id)
 
8
  DEVICE = "cuda"
9
 
10
 
11
+ model_id = "agarkovv/CryptoTrader-LM"
12
  base_model_id = "mistralai/Ministral-8B-Instruct-2410"
13
 
14
  model = AutoPeftModelForCausalLM.from_pretrained(model_id)