ai-forever commited on
Commit
9b21b1b
·
verified ·
1 Parent(s): 19f7397

Update src/kandinsky.py

Browse files
Files changed (1) hide show
  1. 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=10) #[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)
 
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)