nickfraser
commited on
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(
|