Spaces:

yizhezhu
/

MoMA_zeroGPU

Running on Zero

yizhezhu commited on Jun 25

Commit

6688e56

•

1 Parent(s): 545073e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ args = parse_args()
 model = MoMA_main_modal(args).to(device, dtype=torch.float16)
-@spaces.GPU(duration=150)
 def inference(rgb, subject, prompt, strength, seed):
     global model
     seed = int(seed) if seed else 0

 model = MoMA_main_modal(args).to(device, dtype=torch.float16)
+@spaces.GPU(duration=200)
 def inference(rgb, subject, prompt, strength, seed):
     global model
     seed = int(seed) if seed else 0