ffreemt
commited on
Commit
·
d798dc0
1
Parent(s):
2321903
Update
Browse files
app.py
CHANGED
|
@@ -44,7 +44,9 @@ else:
|
|
| 44 |
try:
|
| 45 |
# model = AutoModel.from_pretrained(model_name, trust_remote_code=True).float()
|
| 46 |
model = AutoModelForCausalLM.from_pretrained(
|
| 47 |
-
model_name, trust_remote_code=True
|
|
|
|
|
|
|
| 48 |
).float()
|
| 49 |
except Exception as exc:
|
| 50 |
logger.error(exc)
|
|
|
|
| 44 |
try:
|
| 45 |
# model = AutoModel.from_pretrained(model_name, trust_remote_code=True).float()
|
| 46 |
model = AutoModelForCausalLM.from_pretrained(
|
| 47 |
+
# model_name, trust_remote_code=True
|
| 48 |
+
"model",
|
| 49 |
+
trust_remote_code=True,
|
| 50 |
).float()
|
| 51 |
except Exception as exc:
|
| 52 |
logger.error(exc)
|