Update README.md
Browse files
README.md
CHANGED
|
@@ -156,7 +156,7 @@ quantize(
|
|
| 156 |
)
|
| 157 |
|
| 158 |
quantize(
|
| 159 |
-
pipe.
|
| 160 |
weights=qint8,
|
| 161 |
exclude=[
|
| 162 |
"*.norm", "*.norm1", "*.norm2", "*.norm2_context",
|
|
@@ -166,7 +166,7 @@ quantize(
|
|
| 166 |
|
| 167 |
freeze(pipe.transformer)
|
| 168 |
freeze(pipe.ip_adapter)
|
| 169 |
-
freeze(pipe.
|
| 170 |
|
| 171 |
# Enable Model Offloading
|
| 172 |
pipe.enable_model_cpu_offload()
|
|
|
|
| 156 |
)
|
| 157 |
|
| 158 |
quantize(
|
| 159 |
+
pipe.controlnet,
|
| 160 |
weights=qint8,
|
| 161 |
exclude=[
|
| 162 |
"*.norm", "*.norm1", "*.norm2", "*.norm2_context",
|
|
|
|
| 166 |
|
| 167 |
freeze(pipe.transformer)
|
| 168 |
freeze(pipe.ip_adapter)
|
| 169 |
+
freeze(pipe.controlnet)
|
| 170 |
|
| 171 |
# Enable Model Offloading
|
| 172 |
pipe.enable_model_cpu_offload()
|