Spaces:
Sleeping
Sleeping
ffreemt
commited on
Commit
·
c9a3b27
1
Parent(s):
a296f33
Update model_type='llma'
Browse files
app.py
CHANGED
|
@@ -320,7 +320,7 @@ LLM = AutoModelForCausalLM.from_pretrained(
|
|
| 320 |
# REPO_ID, # DESTINATION_FOLDER, # model_path_or_repo_id: str required
|
| 321 |
# model_file=_,
|
| 322 |
_,
|
| 323 |
-
|
| 324 |
threads=cpu_count,
|
| 325 |
)
|
| 326 |
|
|
|
|
| 320 |
# REPO_ID, # DESTINATION_FOLDER, # model_path_or_repo_id: str required
|
| 321 |
# model_file=_,
|
| 322 |
_,
|
| 323 |
+
model_type="llma", #AutoConfig.from_pretrained(REPO_ID).model_type,
|
| 324 |
threads=cpu_count,
|
| 325 |
)
|
| 326 |
|