commit

app.py

@@ -414,7 +414,6 @@ CHECKPOINTS = ESRGANUpscalerCheckpoints(
 device = DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 DTYPE = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
 enhancer = ESRGANUpscaler(checkpoints=CHECKPOINTS, device=DEVICE, dtype=DTYPE)
-enhancer.to(DEVICE)
 
 # logging
 
@@ -612,7 +611,7 @@ def add_song_cover_text(img,artist,song):
 
     return img
 
-@spaces.GPU(duration=180)
+@spaces.GPU(duration=240)
 def all_pipes(img,pos,neg,artist,song):
     imgs = pipe_generate_image(pos,neg)