Spaces:

fffiloni
/

Artist

Running on Zero

fffiloni commited on Aug 29

Commit

3fa913f

•

1 Parent(s): b32f637

Update injection_main_HF.py

Files changed (1) hide show

injection_main_HF.py CHANGED Viewed

@@ -522,7 +522,7 @@ if __name__ == "__main__":
         # Set up a DDIM scheduler
         pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
-        @spaces.GPU()
         def style_transfer_app(
             prompt,
             image,

         # Set up a DDIM scheduler
         pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
+        @spaces.GPU(duration=90)
         def style_transfer_app(
             prompt,
             image,