Spaces:

vzhizhi6611
/

OmniGen2_x

Running

Sqxww commited on Jun 27

Commit

3a51ba2

1 Parent(s): 061f5b2

reduce gpu duration

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def load_pipeline(accelerator, weight_dtype, args):
         pipeline = pipeline.to(accelerator.device)
     return pipeline
-@spaces.GPU(duration=90)
 def run(
     instruction,
     width_input,

         pipeline = pipeline.to(accelerator.device)
     return pipeline
+@spaces.GPU
 def run(
     instruction,
     width_input,