Sadjad Alikhani
commited on
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -56,7 +56,7 @@ def lwm_inference(preprocessed_chs, input_type, lwm_model):
|
|
| 56 |
embedding_data = embedding_data[:, 1:]
|
| 57 |
|
| 58 |
dataset = embedding_data.float()
|
| 59 |
-
|
| 60 |
return dataset
|
| 61 |
|
| 62 |
def prepare_for_LWM(data, device, batch_size=64, shuffle=False):
|
|
|
|
| 56 |
embedding_data = embedding_data[:, 1:]
|
| 57 |
|
| 58 |
dataset = embedding_data.float()
|
| 59 |
+
print(dataset[0][:4])
|
| 60 |
return dataset
|
| 61 |
|
| 62 |
def prepare_for_LWM(data, device, batch_size=64, shuffle=False):
|