Update modeling_hubert_kmeans.py

modeling_hubert_kmeans.py

@@ -280,7 +280,7 @@ class HubertKmeansModel(PreTrainedModel):
         
         # Return in processor format
         return BatchEncoding({
-            "input_values": outputs.input_ids.float(),  # Return tokens as input_values for compatibility
+            "input_values": outputs.input_ids,  # Return tokens as input_values for compatibility
             "input_ids": outputs.input_ids,
             "attention_mask": inputs.get("attention_mask"),
         })