Merge branch 'main' of https://github.com/hpcaitech/ColossalAI into rlhf_SimPO

This commit is contained in:
YeAnbang
2024-07-10 02:32:07 +00:00
165 changed files with 4783 additions and 4722 deletions

View File

@@ -1,6 +1,7 @@
"""
loss functions
"""
from typing import Optional, Tuple
import torch