yizhangliu commited on
Commit
2db6b53
1 Parent(s): d9c2c8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -97,7 +97,7 @@ sam_predictor = None
97
  sam_mask_generator = None
98
  sd_model = None
99
  lama_cleaner_model= None
100
- lama_cleaner_model_device = device
101
  ram_model = None
102
  kosmos_model = None
103
  kosmos_processor = None
 
97
  sam_mask_generator = None
98
  sd_model = None
99
  lama_cleaner_model= None
100
+ lama_cleaner_model_device = 'cpu' # device
101
  ram_model = None
102
  kosmos_model = None
103
  kosmos_processor = None