Spaces:
Running
on
Zero
Running
on
Zero
delete duration parameter
Browse files- src/kandinsky.py +3 -2
src/kandinsky.py
CHANGED
@@ -34,8 +34,9 @@ pipe = pipe.to(device)
|
|
34 |
|
35 |
MAX_SEED = np.iinfo(np.int32).max
|
36 |
MAX_IMAGE_SIZE = 1024
|
37 |
-
|
38 |
-
@spaces.GPU(duration=12) #[uncomment to use ZeroGPU]
|
|
|
39 |
def generate_image(prompt, width, height, guidance_scale=3, num_inference_steps=50):
|
40 |
|
41 |
# generator = torch.Generator().manual_seed(0)
|
|
|
34 |
|
35 |
MAX_SEED = np.iinfo(np.int32).max
|
36 |
MAX_IMAGE_SIZE = 1024
|
37 |
+
|
38 |
+
# @spaces.GPU(duration=12) #[uncomment to use ZeroGPU]
|
39 |
+
@spaces.GPU()
|
40 |
def generate_image(prompt, width, height, guidance_scale=3, num_inference_steps=50):
|
41 |
|
42 |
# generator = torch.Generator().manual_seed(0)
|