yanze commited on
Commit
6643db0
β€’
1 Parent(s): 9029e51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def get_models(name: str, device: torch.device, offload: bool):
26
  class FluxGenerator:
27
  def __init__(self):
28
  self.device = torch.device('cuda')
29
- self.offload = True
30
  self.model_name = 'flux-dev'
31
  self.model, self.ae, self.t5, self.clip = get_models(
32
  self.model_name,
 
26
  class FluxGenerator:
27
  def __init__(self):
28
  self.device = torch.device('cuda')
29
+ self.offload = False
30
  self.model_name = 'flux-dev'
31
  self.model, self.ae, self.t5, self.clip = get_models(
32
  self.model_name,