ai-forever commited on
Commit
f672b3a
·
verified ·
1 Parent(s): 2793079

Update src/kandinsky.py

Browse files
Files changed (1) hide show
  1. 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, progress=gr.Progress(track_tqdm=True)):
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