tight-inversion-pulid-demo

Running on Zero

tight-inversion commited on 27 days ago

Commit

d672aed

1 Parent(s): 61b1045

Offload

Files changed (1) hide show

app.py CHANGED Viewed

@@ -93,7 +93,7 @@ class FluxGenerator:
         return x
-    @spaces.GPU
     @torch.inference_mode()
     def generate_image(
         self,
@@ -456,5 +456,5 @@ if __name__ == "__main__":
     if args.aggressive_offload:
         args.offload = True
-    demo = create_demo(args, args.name, args.device, args.offload, args.aggressive_offload)
     demo.launch(ssr_mode=False)

         return x
+    @spaces.GPU(duration=90)
     @torch.inference_mode()
     def generate_image(
         self,
     if args.aggressive_offload:
         args.offload = True
+    demo = create_demo(args, args.name, args.device, True, args.aggressive_offload)
     demo.launch(ssr_mode=False)