yanranxiaoxi commited on
Commit
1c843cb
1 Parent(s): b1d6a6c

chore: 指定 CPU 为模型运行环境

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = 'cuda:0'
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)