envs

pipelines/pipeline_imagecoductor.py

@@ -2,6 +2,7 @@
 
 import inspect
 import copy
+import spaces
 from dataclasses import dataclass
 from typing import Any, Callable, Dict, List, Optional, Union
 
@@ -317,6 +318,7 @@ class ImageConductorPipeline(DiffusionPipeline):
         latents = latents * self.scheduler.init_noise_sigma
         return latents
 
+    @spaces.GPU(enable_queue=True, duration=200)
     @torch.no_grad()
     def __call__(
         self,