rename package

src/api.py

@@ -2,8 +2,8 @@ import os
 
 import uvicorn
 
-from llmtuner.api.app import create_app
-from llmtuner.chat import ChatModel
+from llamafactory.api.app import create_app
+from llamafactory.chat import ChatModel
 
 
 def main():

src/llamafactory/data/parser.py

@@ -32,6 +32,8 @@ class DatasetAttr:
     prompt: Optional[str] = "instruction"
     query: Optional[str] = "input"
     response: Optional[str] = "output"
+    chosen: Optional[str] = "chosen"
+    rejected: Optional[str] = "rejected"
     history: Optional[str] = None
     """ columns for the sharegpt format """
     messages: Optional[str] = "conversations"

src/llamafactory/data/utils.py

@@ -10,7 +10,7 @@ if TYPE_CHECKING:
     from datasets import Dataset, IterableDataset
     from transformers import Seq2SeqTrainingArguments
 
-    from llmtuner.hparams import DataArguments
+    from ..hparams import DataArguments
 
 
 logger = get_logger(__name__)

src/llamafactory/extras/misc.py

@@ -30,7 +30,7 @@ except Exception:
 if TYPE_CHECKING:
     from trl import AutoModelForCausalLMWithValueHead
 
-    from llmtuner.hparams import ModelArguments
+    from ..hparams import ModelArguments
 
 
 logger = get_logger(__name__)

src/train.py

@@ -1,4 +1,4 @@
-from llmtuner.train.tuner import run_exp
+from llamafactory.train.tuner import run_exp
 
 
 def main():

src/webui.py

@@ -1,6 +1,6 @@
 import os
 
-from llmtuner.webui.interface import create_ui
+from llamafactory.webui.interface import create_ui
 
 
 def main():