Update inference.py

inference.py

@@ -25,7 +25,7 @@ warnings.filterwarnings('ignore')
 
 def lwm_inference(preprocessed_chs, input_type, lwm_model, device):
     
-    dataset = prepare_for_LWM(preprocessed_chs, device)
+    dataset = prepare_for_lwm(preprocessed_chs, device)
     # Process data through LWM
     lwm_loss, embedding_data = evaluate(lwm_model, dataset)
     print(f'LWM loss: {lwm_loss:.4f}')
@@ -38,7 +38,7 @@ def lwm_inference(preprocessed_chs, input_type, lwm_model, device):
     dataset = embedding_data.float()
     return dataset
 
-def prepare_for_LWM(data, device, batch_size=64, shuffle=False):
+def prepare_for_lwm(data, device, batch_size=64, shuffle=False):
 
     input_ids, masked_tokens, masked_pos = zip(*data)