fffiloni commited on
Commit
b01a9cd
1 Parent(s): defc1c6

trying to get rid off tensors not on same device

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -35,6 +35,7 @@ import gradio as gr
35
  from clip_interrogator import Config, Interrogator
36
 
37
  config = Config()
 
38
  config.blip_offload = True
39
  config.chunk_size = 2048
40
  config.flavor_intermediate_count = 512
35
  from clip_interrogator import Config, Interrogator
36
 
37
  config = Config()
38
+ config.device = 'cuda' if torch.cuda.is_available() else 'cpu'
39
  config.blip_offload = True
40
  config.chunk_size = 2048
41
  config.flavor_intermediate_count = 512