Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
6082621
1
Parent(s):
331f30a
Disable CPU offload as it worsens performance (#4)
Browse files- CPU offload worsens performance (558526aefb4a42004d8dbaae60e2356759e27498)
Co-authored-by: Apolinário from multimodal AI art <multimodalart@users.noreply.huggingface.co>
app.py
CHANGED
@@ -30,7 +30,6 @@ pipe = StableDiffusionXLPipeline(
|
|
30 |
scheduler=scheduler,
|
31 |
force_zeros_for_empty_prompt=False)
|
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)):
|
|
|
30 |
scheduler=scheduler,
|
31 |
force_zeros_for_empty_prompt=False)
|
32 |
pipe = pipe.to("cuda")
|
|
|
33 |
|
34 |
@spaces.GPU(duration=200)
|
35 |
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)):
|