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