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)
|