Spaces:
Sleeping
Sleeping
ffreemt
commited on
Commit
·
e12262f
1
Parent(s):
c9a3b27
Update model_type='llama'
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 |
-
model_type="
|
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="llama", #AutoConfig.from_pretrained(REPO_ID).model_type,
|
324 |
threads=cpu_count,
|
325 |
)
|
326 |
|