nroggendorff commited on
Commit
98b968b
·
verified ·
1 Parent(s): 9f53d90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
6
  prior_pipeline = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.float16).to("cuda")
7
  decoder_pipeline = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.float16).to("cuda")
8
 
9
- @spaces.GPU(duration=480)
10
  def generate(prompt, negative_prompt, width, height, steps):
11
  prior_output = prior_pipeline(
12
  prompt=prompt,
 
6
  prior_pipeline = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.float16).to("cuda")
7
  decoder_pipeline = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.float16).to("cuda")
8
 
9
+ @spaces.GPU
10
  def generate(prompt, negative_prompt, width, height, steps):
11
  prior_output = prior_pipeline(
12
  prompt=prompt,