Spaces:
Runtime error
Runtime error
lorocksUMD
commited on
Commit
•
77ceba1
1
Parent(s):
243d710
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_map="cpu"
|
85 |
)
|
86 |
|
87 |
qs = args.query
|