magenta-js
magenta-js copied to clipboard
checkpoint_converter.py | error in function write_weights ...?
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?
Yes, had the same issue. Changed to 'quantization_dtype_map' and worked fine. Opened a new pull request as well.