Spaces:
Runtime error
Runtime error
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 |
|