Offload

app.py

@@ -93,7 +93,7 @@ class FluxGenerator:
 
         return x
 
-    @spaces.GPU
+    @spaces.GPU(duration=90)
     @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 = create_demo(args, args.name, args.device, True, args.aggressive_offload)
     demo.launch(ssr_mode=False)