fix: remove streaming from output path name
This commit is contained in:
@@ -90,7 +90,7 @@ def streaming_quantize(model_path: str, output_path: str):
|
|||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
parser.add_argument("--model-path", type=str, default="/data/models/Ornith-1.0-35B")
|
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-streaming")
|
parser.add_argument("--output-path", type=str, default="/data/models/Ornith-1.0-35B-nf4")
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
streaming_quantize(args.model_path, args.output_path)
|
streaming_quantize(args.model_path, args.output_path)
|
||||||
|
|||||||
Reference in New Issue
Block a user