This commit is contained in:
hiyouga
2023-10-10 17:41:13 +08:00
parent 8e2ed6b8ce
commit e1dcb8e4dc
9 changed files with 22 additions and 29 deletions

View File

@@ -1,11 +1,12 @@
import gradio as gr
import logging
import os
import threading
import time
import logging
import threading
import gradio as gr
from typing import Any, Dict, Generator, List, Tuple
import transformers
from transformers.trainer import TRAINING_ARGS_NAME
from typing import Any, Dict, Generator, List, Tuple
from llmtuner.extras.callbacks import LogCallback
from llmtuner.extras.constants import DEFAULT_MODULE, TRAINING_STAGES