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,
|