nickfraser commited on
Commit
77d84b5
1 Parent(s): e6b40f3

Fix (script): Quantize output of QuantLora layers.

Browse files
Files changed (1) hide show
  1. 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, quant_layer_kwargs),
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(