crystal-technologies commited on
Commit
2c6c5df
1 Parent(s): b6c3744

Update Perceptrix/engine.py

Browse files
Files changed (1) hide show
  1. Perceptrix/engine.py +1 -1
Perceptrix/engine.py CHANGED
@@ -8,7 +8,7 @@ import os
8
 
9
  model_name = os.environ.get('LLM_MODEL')
10
 
11
- model_path = "models/CRYSTAL-instruct" if model_name != None else model_name
12
 
13
  config = transformers.AutoConfig.from_pretrained(
14
  model_name, trust_remote_code=True)
 
8
 
9
  model_name = os.environ.get('LLM_MODEL')
10
 
11
+ model_path = "models/CRYSTAL-instruct" if model_name == None else model_name
12
 
13
  config = transformers.AutoConfig.from_pretrained(
14
  model_name, trust_remote_code=True)