Spaces:
Running
on
Zero
Running
on
Zero
Update src/kandinsky.py
Browse files- src/kandinsky.py +1 -1
src/kandinsky.py
CHANGED
@@ -35,7 +35,7 @@ pipe = pipe.to(device)
|
|
35 |
MAX_SEED = np.iinfo(np.int32).max
|
36 |
MAX_IMAGE_SIZE = 1024
|
37 |
|
38 |
-
@spaces.GPU(duration=
|
39 |
def generate_image(prompt, width, height, guidance_scale=3, num_inference_steps=50):
|
40 |
|
41 |
# generator = torch.Generator().manual_seed(0)
|
|
|
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)
|