Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Refactor
Browse files Browse the repository at this point in the history
Signed-off-by: Austin Liu <austin362667@gmail.com>
austin362667 committed Nov 14, 2024

Verified

This commit was signed with the committer’s verified signature.
austin362667 Austin Liu
1 parent 0e71106 commit b88708d
Showing 5 changed files with 343 additions and 376 deletions.
6 changes: 4 additions & 2 deletions benchmark/scripts/benchmark_dpo_loss.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
from test.chunked_loss.test_dpo_loss import HF_DPO_Loss

import torch
import triton
from utils import (
QUANTILES,
SingleBenchmarkRunInput,
@@ -9,7 +10,8 @@
run_benchmarks,
)

from liger_kernel.alignment.dpo_loss import HF_DPO_Loss, LigerFusedLinearDPOFunction
import triton
from liger_kernel.chunked_loss.dpo_loss import LigerFusedLinearDPOFunction


class TorchDPOLoss(torch.nn.Module):
301 changes: 0 additions & 301 deletions src/liger_kernel/alignment/dpo_loss.py

This file was deleted.

Loading

0 comments on commit b88708d

Please sign in to comment.