Spaces:

tight-inversion
/

tight-inversion-pulid-demo

Running on Zero

tight-inversion commited on 4 days ago

Commit

1a8a74e

1 Parent(s): b785460

Fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,7 +34,7 @@ class FluxGenerator:
         self.device = torch.device('cuda')
         self.offload = False
         self.model_name = 'flux-dev'
-        self.model, self.ae, self.t5, self.clip  = get_models(
             self.model_name,
             device=self.device,
             offload=self.offload,

         self.device = torch.device('cuda')
         self.offload = False
         self.model_name = 'flux-dev'
+        self.model, self.ae, self.t5, self.clip_model  = get_models(
             self.model_name,
             device=self.device,
             offload=self.offload,