Remove unused import

Signed-off-by: Mustafa Eyceoz <meyceoz@redhat.com>
instructlab · Jun 19, 2024 · 997ad36 · 997ad36
1 parent 784e2d4
commit 997ad36
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/instructlab/training/token_dataset.py b/src/instructlab/training/token_dataset.py
@@ -6,7 +6,6 @@
 from torch.utils.data import DataLoader, Dataset
 import numpy as np
 import torch
-import torch.nn.functional as F
 
 # First Party
 from instructlab.training.multipack_sampler import MultipackDistributedBatchSampler