Update app.py
Browse files
app.py
CHANGED
|
@@ -56,9 +56,9 @@ high_threshold = 200
|
|
| 56 |
|
| 57 |
pipe, params = load_sb_pipe(controlnet_version)
|
| 58 |
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
print("Loaded models...")
|
| 63 |
def pipe_inference(
|
| 64 |
image,
|
|
|
|
| 56 |
|
| 57 |
pipe, params = load_sb_pipe(controlnet_version)
|
| 58 |
|
| 59 |
+
pipe.enable_xformers_memory_efficient_attention()
|
| 60 |
+
pipe.enable_model_cpu_offload()
|
| 61 |
+
pipe.enable_attention_slicing()
|
| 62 |
print("Loaded models...")
|
| 63 |
def pipe_inference(
|
| 64 |
image,
|