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

fix #3317

Browse Source
This commit is contained in:
hiyouga
2024-04-17 22:17:19 +08:00
parent e3ca57602c
commit 6d641af703
15 changed files with 75 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/llmtuner/webui/common.py
View File

@@ -3,7 +3,6 @@ import os
from collections import defaultdict
from typing import Any, Dict, Optional
import gradio as gr
from peft.utils import SAFETENSORS_WEIGHTS_NAME, WEIGHTS_NAME
from ..extras.constants import (
@@ -17,6 +16,11 @@ from ..extras.constants import (
DownloadSource,
)
from ..extras.misc import use_modelscope
from ..extras.packages import is_gradio_available
if is_gradio_available():
import gradio as gr
ADAPTER_NAMES = {WEIGHTS_NAME, SAFETENSORS_WEIGHTS_NAME}