Skip to content

Commit

Permalink
Update export.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Giuseppe5 authored Jun 21, 2024
1 parent f431044 commit 6542813
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/brevitas_examples/stable_diffusion/sd_quant/export.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ def handle_quant_param(layer, layer_dict):
layer_dict['weight_scale_shape'] = weight_scale_shape
layer_dict['weight_zp'] = weight_zp.numpy().tolist()
layer_dict['weight_zp_shape'] = weight_scale_shape
layer_dict['weight_zp_dtype'] = str(torch.uint8)
layer_dict['weight_zp_dtype'] = str(torch.int8)
return layer_dict


Expand Down

0 comments on commit 6542813

Please sign in to comment.