Spaces:

p1atdev
/

danbooru-tags-transformer-v2

Running on Zero

p1atdev commited on May 5, 2024

Commit

17aaee2

1 Parent(s): 284b24d

chore: remove zero duration

Files changed (1) hide show

diffusion.py CHANGED Viewed

@@ -95,7 +95,7 @@ class ImageGenerator:
         self.pipe.to(device)
-    @spaces.GPU(duration=30)
     def generate(
         self,
         prompt: str,

         self.pipe.to(device)
+    @spaces.GPU()
     def generate(
         self,
         prompt: str,