diff --git a/src/axolotl/utils/dataloader.py b/src/axolotl/utils/dataloader.py
index f19232625..77cc933b3 100644
--- a/src/axolotl/utils/dataloader.py
+++ b/src/axolotl/utils/dataloader.py
@@ -99,6 +99,7 @@ def allocate(
 
 class MultipackDistributedDataloader:
     """Unpadded data loading using Multipack.
+    Adapted from https://github.com/imoneoi/openchat/blob/v3_fix_mle_loss/ochat/training_deepspeed/multipack_dataloader.py
     Approximate (at most ~1.22x) the optimal solution of the identical-machines scheduling problem, which is NP-hard.
     """