diff --git a/ml/src/tft/quantized_attention.rs b/ml/src/tft/quantized_attention.rs
index 6457fe563..36eae00ad 100644
--- a/ml/src/tft/quantized_attention.rs
+++ b/ml/src/tft/quantized_attention.rs
@@ -9,7 +9,7 @@ use crate::memory_optimization::quantization::{
     QuantizationConfig, QuantizationType, QuantizedTensor, Quantizer,
 };
 use crate::MLError;
-use candle_core::{DType, Device, Tensor};
+use candle_core::{Device, Tensor};
 use candle_nn::VarBuilder;
 use std::collections::HashMap;