nroggendorff commited on
Commit
ef8be4d
·
verified ·
1 Parent(s): 265014b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ decoder_pipeline = StableCascadeDecoderPipeline.from_pretrained("stabilityai/sta
9
  prior_pipeline.enable_model_cpu_offload()
10
  decoder_pipeline.enable_model_cpu_offload()
11
 
12
- prior = prior_pipeline#.to("cuda")
13
- decoder = decoder_pipeline#.to("cuda")
14
 
15
  @spaces.GPU
16
  def generate(prompt, negative_prompt, steps):
 
9
  prior_pipeline.enable_model_cpu_offload()
10
  decoder_pipeline.enable_model_cpu_offload()
11
 
12
+ prior = prior_pipeline.to("cuda")
13
+ decoder = decoder_pipeline.to("cuda")
14
 
15
  @spaces.GPU
16
  def generate(prompt, negative_prompt, steps):