ai-forever commited on
Commit
506c81c
·
verified ·
1 Parent(s): a963159

Update src/kandinsky.py

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