Spaces:

yizhezhu
/

MoMA_zeroGPU

Configuration error

yizhezhu commited on Jan 13

Commit

6257d55

verified ·

1 Parent(s): 225005c

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=200)
 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=120)
 def inference(rgb, subject, prompt, strength, seed):
     global model
     seed = int(seed) if seed else 0