Test

app.py

@@ -20,18 +20,18 @@ from pulid.pipeline_flux import PuLIDPipeline
 from pulid.utils import resize_numpy_image_long, seed_everything
 
 
-def get_models(name: str, device: torch.device, offload: bool, fp8: bool):
-    t5 = load_t5(device, max_length=128)
-    clip = load_clip(device)
-    model = load_flow_model(name, device="cpu" if offload else device)
-    model.eval()
-    ae = load_ae(name, device=device)
-    return model, ae, t5, clip
+# def get_models(name: str, device: torch.device, offload: bool, fp8: bool):
+#     t5 = load_t5(device, max_length=128)
+#     clip = load_clip(device)
+#     model = load_flow_model(name, device="cpu" if offload else device)
+#     model.eval()
+#     ae = load_ae(name, device=device)
+#     return model, ae, t5, clip
+class Tmp:
+    def __init__(self):
+        self.max_length = 128
 
 def get_models(name: str, device: torch.device, offload: bool, fp8: bool):
-    class Tmp:
-        def __init__(self):
-            self.max_length = 128
     t5 = Tmp()
     clip = None
     model = None