svjack commited on
Commit
f2837ed
·
verified ·
1 Parent(s): 8f86882

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -1,3 +1,7 @@
 
 
 
 
1
  import os
2
  import re
3
  import time
@@ -46,7 +50,8 @@ def encode(init_image, 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)
 
1
+ '''
2
+ huggingface-cli login
3
+ '''
4
+
5
  import os
6
  import re
7
  import time
 
50
  return init_image
51
 
52
  torch_device = "cuda" if torch.cuda.is_available() else "cpu"
53
+ #offload = False
54
+ offload = True
55
  device = "cuda" if torch.cuda.is_available() else "cpu"
56
  name = 'flux-dev'
57
  ae = load_ae(name, device="cpu" if offload else torch_device)