crystal-technologies
commited on
Commit
•
2c6c5df
1
Parent(s):
b6c3744
Update Perceptrix/engine.py
Browse files- 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
|
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)
|