nroggendorff commited on
Commit
5b81cb7
·
verified ·
1 Parent(s): 80a0c21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from diffusers import FluxPipeline
7
  pipeline = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16).to("cuda")
8
  #pipeline.enable_model_cpu_offload()
9
 
10
- @spaces.GPU(duration=190)
11
  def generate(prompt, negative_prompt, width, height, sample_steps):
12
  return pipeline(prompt=f"{prompt}\nDO NOT INCLUDE {negative_prompt}", width=width, height=height, num_inference_steps=sample_steps, generator=torch.Generator("cpu").manual_seed(42), guidance_scale=7).images[0]
13
 
 
7
  pipeline = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16).to("cuda")
8
  #pipeline.enable_model_cpu_offload()
9
 
10
+ @spaces.GPU(duration=70)
11
  def generate(prompt, negative_prompt, width, height, sample_steps):
12
  return pipeline(prompt=f"{prompt}\nDO NOT INCLUDE {negative_prompt}", width=width, height=height, num_inference_steps=sample_steps, generator=torch.Generator("cpu").manual_seed(42), guidance_scale=7).images[0]
13