Update app.py
Browse files
app.py
CHANGED
@@ -39,9 +39,7 @@ if has_cuda:
|
|
39 |
# offload_folder="offload_folder",
|
40 |
).cuda()
|
41 |
else:
|
42 |
-
model = (
|
43 |
-
AutoModel.from_pretrained(model_name, trust_remote_code=True).float()
|
44 |
-
).float()
|
45 |
|
46 |
model = model.eval()
|
47 |
|
|
|
39 |
# offload_folder="offload_folder",
|
40 |
).cuda()
|
41 |
else:
|
42 |
+
model = AutoModel.from_pretrained(model_name, trust_remote_code=True).float()
|
|
|
|
|
43 |
|
44 |
model = model.eval()
|
45 |
|