Fix train_data_file usage

app.py

@@ -21,6 +21,7 @@ class QuantizationConfig:
     method: str
     use_imatrix: bool = False
     imatrix_method: str = "IQ4_NL"
+    train_data: str = ""
     quant_embedding: bool = False
     embedding_tensor_method: str = "Q8_0"
     leave_output: bool = False
@@ -149,7 +150,11 @@ class HuggingFaceModelProcessor:
         if not os.path.isfile(quant_config.fp16_model):
             raise GGUFConverterError(f"Model file not found: {quant_config.fp16_model}")
 
-        train_data_path = self.CALIBRATION_FILE
+        if quant_config.train_data:
+            train_data_path = quant_config.train_data
+        else:
+            train_data_path = self.CALIBRATION_FILE
+
         if not os.path.isfile(train_data_path):
             raise GGUFConverterError(f"Training data file not found: {train_data_path}")
 
@@ -798,6 +803,7 @@ class GGUFConverterUI:
                 method=q_method,
                 use_imatrix=use_imatrix,
                 imatrix_method=imatrix_q_method,
+                train_data=train_data_file.name,
                 quant_embedding=quant_embedding,
                 embedding_tensor_method=embedding_tensor_method,
                 leave_output=leave_output,