Spaces:

dgoot
/

image-to-image

Runtime error

dgoot commited on Sep 30, 2024

Commit

ef0ba0f

1 Parent(s): 2e98974

Set GPU duration to 3 min

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ models = [
 @logger.catch(reraise=True)
-@spaces.GPU
 def generate(
     model: str,
     prompt: str,
@@ -28,7 +28,7 @@ def generate(
     init_image.thumbnail((1024, 1024))
     def progress_callback(pipe, step_index, timestep, callback_kwargs):
-        logger.debug(
             f"Callback: {dict(num_timesteps=pipe.num_timesteps, step_index=step_index, timestep=timestep)}"
         )
         progress((step_index + 1, pipe.num_timesteps))

 @logger.catch(reraise=True)
+@spaces.GPU(duration=180)
 def generate(
     model: str,
     prompt: str,
     init_image.thumbnail((1024, 1024))
     def progress_callback(pipe, step_index, timestep, callback_kwargs):
+        logger.trace(
             f"Callback: {dict(num_timesteps=pipe.num_timesteps, step_index=step_index, timestep=timestep)}"
         )
         progress((step_index + 1, pipe.num_timesteps))