Former-commit-id: 31ccd3329ac634b239c43d60bd955cd95670df16
This commit is contained in:
hiyouga
2023-10-10 17:41:13 +08:00
parent 820540780a
commit 6d0d46c7fb
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