tight-inversion commited on
Commit
b785460
·
1 Parent(s): c67f10c
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -22,11 +22,11 @@ from pulid.utils import resize_numpy_image_long, seed_everything
22
 
23
  def get_models(name: str, device: torch.device, offload: bool):
24
  t5 = load_t5(device, max_length=128)
25
- clip = load_clip(device)
26
  model = load_flow_model(name, device="cpu" if offload else device)
27
  model.eval()
28
  ae = load_ae(name, device="cpu" if offload else device)
29
- return model, ae, t5, clip
30
 
31
 
32
  class FluxGenerator:
@@ -46,7 +46,7 @@ class FluxGenerator:
46
  flux_generator = FluxGenerator()
47
 
48
 
49
- @spaces.GPU(duration=30)
50
  @torch.inference_mode()
51
  def generate_image(
52
  prompt: str,
 
22
 
23
  def get_models(name: str, device: torch.device, offload: bool):
24
  t5 = load_t5(device, max_length=128)
25
+ clip_model = load_clip(device)
26
  model = load_flow_model(name, device="cpu" if offload else device)
27
  model.eval()
28
  ae = load_ae(name, device="cpu" if offload else device)
29
+ return model, ae, t5, clip_model
30
 
31
 
32
  class FluxGenerator:
 
46
  flux_generator = FluxGenerator()
47
 
48
 
49
+ @spaces.GPU(duration=80)
50
  @torch.inference_mode()
51
  def generate_image(
52
  prompt: str,