fix save string

Author: dsikka

examples/quantization_w4a4_fp4/trinity_large_nvfp4.py

@@ -21,7 +21,7 @@
 DATASET_ID = "HuggingFaceH4/ultrachat_200k"
 DATASET_SPLIT = "train_sft"
 
-NUM_CALIBRATION_SAMPLES = 100
+NUM_CALIBRATION_SAMPLES = 256
 MAX_SEQUENCE_LENGTH = 2048
 
 # Load dataset and preprocess.
@@ -89,6 +89,6 @@ def tokenize(sample):
 
 
 # Save to disk in compressed-tensors format.
-SAVE_DIR = "/raid/engine/dsikka/" + MODEL_ID.rstrip("/").split("/")[-1] + "-NVFP4"
-model.save_pretrained(SAVE_DIR, save_compressed=True)
+SAVE_DIR = MODEL_ID.rstrip("/").split("/")[-1] + "-NVFP4"
+model.save_pretrained(SAVE_DIR)
 tokenizer.save_pretrained(SAVE_DIR)