diff --git a/scripts/tensorize_model.py b/scripts/tensorize_model.py index 1a2eb02..f1f68ad 100755 --- a/scripts/tensorize_model.py +++ b/scripts/tensorize_model.py @@ -58,12 +58,7 @@ def tensorize_model( serializer.write_module(model) serializer.close() - # Write config to tensorized model weights directory - dir_path = os.path.dirname(tensorizer_path) - config_path = os.path.join(dir_path, 'config.json') - model_config.save_pretrained(config_path) - - logger.info(f"Tensorized model {model_name} in {dtype} and wrote tensors to {tensorizer_path} and config to {config_path}...") + logger.info(f"Tensorized model {model_name} in {dtype} and wrote tensors to {tensorizer_path}...") return {"tensorized_weights_path": tensorizer_path, "dtype": dtype} @@ -87,4 +82,4 @@ def tensorize_model( model_path=args.model_path, tensorizer_path=args.tensorizer_path, dtype=args.dtype - ) \ No newline at end of file + )