Add DoRA type support. move VALID_LORA_TYPES to utils.constants and update imports in recipe parsers and API routes.

This commit is contained in:
Will Miao
2025-06-12 09:25:00 +08:00
parent 26d105c439
commit 78cac2edc2
4 changed files with 11 additions and 8 deletions

View File

@@ -7,7 +7,7 @@ import re
from typing import Dict, List, Any, Optional, Tuple
from abc import ABC, abstractmethod
from ..config import config
from .constants import VALID_LORA_TYPES
from ..utils.constants import VALID_LORA_TYPES
logger = logging.getLogger(__name__)

View File

@@ -1,5 +1,8 @@
"""Constants used across recipe parsers."""
# Import VALID_LORA_TYPES from utils.constants
from ..utils.constants import VALID_LORA_TYPES
# Constants for generation parameters
GEN_PARAM_KEYS = [
'prompt',
@@ -11,6 +14,3 @@ GEN_PARAM_KEYS = [
'size',
'clip_skip',
]
# Valid Lora types
VALID_LORA_TYPES = ['lora', 'locon']