Former-commit-id: 7dce1763be4374cf616d96db95ae964ff510a9d6
This commit is contained in:
hiyouga
2024-04-17 22:17:19 +08:00
parent 171a029c5e
commit 9f6349a333
15 changed files with 75 additions and 32 deletions

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}