Spaces:

comrender
/

fluxhdupscaler

Running on Zero

comrender commited on 2 days ago

Commit

776edd6

verified ·

1 Parent(s): 353e9dc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -162,7 +162,7 @@ def enhance_image(
                 "black-forest-labs/FLUX.1-dev",
                 torch_dtype=dtype,
                 low_cpu_mem_usage=True,
-                device_map="auto"
             )
             pipe.enable_vae_tiling()
             pipe.enable_vae_slicing()
@@ -191,7 +191,7 @@ def enhance_image(
                 "black-forest-labs/FLUX.1-dev",
                 torch_dtype=dtype,
                 low_cpu_mem_usage=True,
-                device_map="auto"
             )
             pipe.enable_vae_tiling()
             pipe.enable_vae_slicing()

                 "black-forest-labs/FLUX.1-dev",
                 torch_dtype=dtype,
                 low_cpu_mem_usage=True,
+                device_map="balanced"
             )
             pipe.enable_vae_tiling()
             pipe.enable_vae_slicing()
                 "black-forest-labs/FLUX.1-dev",
                 torch_dtype=dtype,
                 low_cpu_mem_usage=True,
+                device_map="balanced"
             )
             pipe.enable_vae_tiling()
             pipe.enable_vae_slicing()