423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-15 03:10:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix modelscope data hub

Browse Source
This commit is contained in:
hiyouga
2023-12-12 18:33:06 +08:00
parent 382319915c
commit d5b2c57a35
6 changed files with 60 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/model/loader.py
View File

@@ -1,4 +1,3 @@
import os
import math
import torch
from types import MethodType
@@ -13,7 +12,6 @@ from transformers import (
PreTrainedModel,
PreTrainedTokenizerBase
)
from transformers.models.llama import modeling_llama as LlamaModule
from transformers.utils.versions import require_version
from trl import AutoModelForCausalLMWithValueHead