magenta-js icon indicating copy to clipboard operation
magenta-js copied to clipboard

checkpoint_converter.py | error in function write_weights ...?

Open Oriolla opened this issue 3 years ago • 1 comments

In checkpoint_converter.py:

write_weights([entries], output_dir, write_manifest=True, quantization_dtype=quantization_dtype, shard_size_bytes=shard_mb 1024 1024)

Is it right? May be quantization_dtype_map?

Oriolla avatar Jun 03 '21 21:06 Oriolla

Yes, had the same issue. Changed to 'quantization_dtype_map' and worked fine. Opened a new pull request as well.

damjanprvc avatar Mar 25 '22 03:03 damjanprvc