Spaces:
Runtime error
Runtime error
yanranxiaoxi
commited on
Commit
•
1c843cb
1
Parent(s):
b1d6a6c
chore: 指定 CPU 为模型运行环境
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ parser.add_argument("--load_8bit_llava", action='store_true', default=False)
|
|
43 |
args = parser.parse_args()
|
44 |
|
45 |
if torch.cuda.device_count() > 0:
|
46 |
-
BOOXEL_device = '
|
47 |
|
48 |
# 加载 BOOXEL
|
49 |
model, default_setting = create_BOOXEL_model(args.opt, BOOXEL_sign='Q', load_default_setting=True)
|
|
|
43 |
args = parser.parse_args()
|
44 |
|
45 |
if torch.cuda.device_count() > 0:
|
46 |
+
BOOXEL_device = 'cpu'
|
47 |
|
48 |
# 加载 BOOXEL
|
49 |
model, default_setting = create_BOOXEL_model(args.opt, BOOXEL_sign='Q', load_default_setting=True)
|