ai-forever commited on
Commit
ed529e5
·
verified ·
1 Parent(s): bc56903

delete duration parameter

Browse files
Files changed (1) hide show
  1. src/kandinsky.py +3 -2
src/kandinsky.py CHANGED
@@ -34,8 +34,9 @@ pipe = pipe.to(device)
34
 
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)
 
34
 
35
  MAX_SEED = np.iinfo(np.int32).max
36
  MAX_IMAGE_SIZE = 1024
37
+
38
+ # @spaces.GPU(duration=12) #[uncomment to use ZeroGPU]
39
+ @spaces.GPU()
40
  def generate_image(prompt, width, height, guidance_scale=3, num_inference_steps=50):
41
 
42
  # generator = torch.Generator().manual_seed(0)