kyy/llm_trainer
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update packing

Browse Source
This commit is contained in:
hiyouga committed 2024-07-04 01:10:55 +08:00
1 parent a36e8f2dd5
commit cce7083024
6 files changed
+133 -271
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/llamafactory/extras/constants.py
-3
View File
@@ -82,14 +82,11 @@ SUPPORTED_CLASS_FOR_BLOCK_DIAG_ATTN = {
"falcon",
"gemma",
"gemma2",
"jamba",
"llama",
"mistral",
"mixtral",
"phi",
"phi3",
"qwen2",
"qwen2_moe",
"starcoder2",
}