MagicBag commited on
Commit
e09baa5
1 Parent(s): 12b0bf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def encode(init_image, torch_device):
46
  return init_image
47
 
48
  torch_device = "cuda" if torch.cuda.is_available() else "cpu"
49
- offload = True
50
  device = "cuda" if torch.cuda.is_available() else "cpu"
51
  name = 'flux-dev'
52
  ae = load_ae(name, device="cpu" if offload else torch_device)
 
46
  return init_image
47
 
48
  torch_device = "cuda" if torch.cuda.is_available() else "cpu"
49
+ offload = False
50
  device = "cuda" if torch.cuda.is_available() else "cpu"
51
  name = 'flux-dev'
52
  ae = load_ae(name, device="cpu" if offload else torch_device)