Hugging Face's logo

Spaces:
gokaygokay
/
Kolors
Running on Zero

App Files Community
6
gokaygokay commited on
Commit
331f30a
•
1 Parent(s): 29f3401

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ pipe = StableDiffusionXLPipeline(
32
  pipe = pipe.to("cuda")
33
  pipe.enable_model_cpu_offload()
34
 
35
- @spaces.GPU
36
  def generate_image(prompt, negative_prompt, height, width, num_inference_steps, guidance_scale, num_images_per_prompt, use_random_seed, seed, progress=gr.Progress(track_tqdm=True)):
37
  if use_random_seed:
38
  seed = random.randint(0, 2**32 - 1)
 
32
  pipe = pipe.to("cuda")
33
  pipe.enable_model_cpu_offload()
34
 
35
+ @spaces.GPU(duration=200)
36
  def generate_image(prompt, negative_prompt, height, width, num_inference_steps, guidance_scale, num_images_per_prompt, use_random_seed, seed, progress=gr.Progress(track_tqdm=True)):
37
  if use_random_seed:
38
  seed = random.randint(0, 2**32 - 1)