minor update

Signed-off-by: Yu Chin Fabian Lim <[email protected]>
foundation-model-stack · Aug 16, 2024 · 1d3f481 · 1d3f481
1 parent 3dbdde5
commit 1d3f481
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/plugins/attention-and-distributed-packing/src/fms_acceleration_aadp/multipack_sampler.py b/plugins/attention-and-distributed-packing/src/fms_acceleration_aadp/multipack_sampler.py
@@ -20,7 +20,8 @@
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
-taken from https://github.com/imoneoi/multipack_sampler
+taken from https://github.com/imoneoi/multipack_sampler with some modifications 
+taken from https://github.com/instructlab/training/blob/main/src/instructlab/training/multipack_sampler.py
 """
 
 # Standard