fffiloni commited on
Commit
f5611af
1 Parent(s): 4b2f815

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ from clip_interrogator import Config, Interrogator
37
 
38
  config = Config()
39
  config.device = 'cuda' if torch.cuda.is_available() else 'cpu'
40
- config.blip_offload = True
41
  config.chunk_size = 2048
42
  config.flavor_intermediate_count = 512
43
  config.blip_num_beams = 64
 
37
 
38
  config = Config()
39
  config.device = 'cuda' if torch.cuda.is_available() else 'cpu'
40
+ config.blip_offload = False if torch.cuda.is_available() else True
41
  config.chunk_size = 2048
42
  config.flavor_intermediate_count = 512
43
  config.blip_num_beams = 64