Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ from diffusers import DiffusionPipeline
|
|
| 7 |
pipeline = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev").to("cuda")
|
| 8 |
pipeline.enable_model_cpu_offload()
|
| 9 |
|
| 10 |
-
@spaces.GPU(duration=120)
|
| 11 |
def generate(prompt, negative_prompt, width, height, sample_steps):
|
| 12 |
return pipeline(prompt=f"{prompt}\nDO NOT INCLUDE {negative_prompt} FOR ANY REASON", width=width, height=height, num_inference_steps=sample_steps, generator=torch.Generator("cpu").manual_seed(127), guidance_scale=7).images[0]
|
| 13 |
|
|
|
|
| 7 |
pipeline = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev").to("cuda")
|
| 8 |
pipeline.enable_model_cpu_offload()
|
| 9 |
|
| 10 |
+
@spaces.GPU#(duration=120)
|
| 11 |
def generate(prompt, negative_prompt, width, height, sample_steps):
|
| 12 |
return pipeline(prompt=f"{prompt}\nDO NOT INCLUDE {negative_prompt} FOR ANY REASON", width=width, height=height, num_inference_steps=sample_steps, generator=torch.Generator("cpu").manual_seed(127), guidance_scale=7).images[0]
|
| 13 |
|