Spaces:
Running
on
Zero
Running
on
Zero
tight-inversion
commited on
Commit
·
b785460
1
Parent(s):
c67f10c
Fix
Browse files
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 |
-
|
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,
|
30 |
|
31 |
|
32 |
class FluxGenerator:
|
@@ -46,7 +46,7 @@ class FluxGenerator:
|
|
46 |
flux_generator = FluxGenerator()
|
47 |
|
48 |
|
49 |
-
@spaces.GPU(duration=
|
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,
|