fix: use NF4 naming convention for quantized model
This commit is contained in:
@@ -41,7 +41,7 @@ if __name__ == "__main__":
|
||||
import argparse
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("--model-path", type=str, default="/data/models/Ornith-1.0-35B")
|
||||
parser.add_argument("--output-path", type=str, default="/data/models/Ornith-1.0-35B-4bit-nf4")
|
||||
parser.add_argument("--output-path", type=str, default="/data/models/Ornith-1.0-35B-nf4")
|
||||
args = parser.parse_args()
|
||||
|
||||
quantize_model(args.model_path, args.output_path)
|
||||
|
||||
Reference in New Issue
Block a user