Spaces:
Running
on
Zero
Running
on
Zero
Update src/kandinsky.py
Browse files- src/kandinsky.py +2 -2
src/kandinsky.py
CHANGED
@@ -36,7 +36,7 @@ MAX_SEED = np.iinfo(np.int32).max
|
|
36 |
MAX_IMAGE_SIZE = 1024
|
37 |
|
38 |
@spaces.GPU #[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)
|
42 |
|
@@ -48,7 +48,7 @@ def generate_image(prompt, width, height, guidance_scale=3, num_inference_steps=
|
|
48 |
height = height,
|
49 |
generator = generator
|
50 |
).images[0]
|
51 |
-
|
52 |
return image
|
53 |
|
54 |
|
|
|
36 |
MAX_IMAGE_SIZE = 1024
|
37 |
|
38 |
@spaces.GPU #[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)
|
42 |
|
|
|
48 |
height = height,
|
49 |
generator = generator
|
50 |
).images[0]
|
51 |
+
|
52 |
return image
|
53 |
|
54 |
|