Spaces:
Runtime error
Runtime error
lorocksUMD
commited on
Commit
•
243d710
1
Parent(s):
19ba12a
Update app.py
Browse files
app.py
CHANGED
@@ -81,7 +81,7 @@ args = type('Args', (), {
|
|
81 |
|
82 |
|
83 |
tokenizer, model, image_processor, context_len = load_pretrained_model(
|
84 |
-
model_path, None, model_name
|
85 |
)
|
86 |
|
87 |
qs = args.query
|
|
|
81 |
|
82 |
|
83 |
tokenizer, model, image_processor, context_len = load_pretrained_model(
|
84 |
+
model_path, None, model_name, device_name="cpu"
|
85 |
)
|
86 |
|
87 |
qs = args.query
|