Commit ·
77d84b5
1
Parent(s): e6b40f3
Fix (script): Quantize output of QuantLora layers.
Browse files- minimal_script.py +1 -1
minimal_script.py
CHANGED
|
@@ -177,7 +177,7 @@ def main(args):
|
|
| 177 |
nn.Linear: (qnn.QuantLinear, quant_linear_kwargs),
|
| 178 |
nn.Conv2d: (qnn.QuantConv2d, quant_layer_kwargs),
|
| 179 |
'diffusers.models.lora.LoRACompatibleLinear':
|
| 180 |
-
(LoRACompatibleQuantLinear,
|
| 181 |
'diffusers.models.lora.LoRACompatibleConv': (LoRACompatibleQuantConv2d, quant_layer_kwargs)}
|
| 182 |
|
| 183 |
pipe.unet = layerwise_quantize(
|
|
|
|
| 177 |
nn.Linear: (qnn.QuantLinear, quant_linear_kwargs),
|
| 178 |
nn.Conv2d: (qnn.QuantConv2d, quant_layer_kwargs),
|
| 179 |
'diffusers.models.lora.LoRACompatibleLinear':
|
| 180 |
+
(LoRACompatibleQuantLinear, quant_linear_kwargs),
|
| 181 |
'diffusers.models.lora.LoRACompatibleConv': (LoRACompatibleQuantConv2d, quant_layer_kwargs)}
|
| 182 |
|
| 183 |
pipe.unet = layerwise_quantize(
|