Skip to content

Commit f5177b0

Browse files
committed
Added new params to support Fish Audio
1 parent d20aaa8 commit f5177b0

1 file changed

Lines changed: 25 additions & 4 deletions

File tree

runware/types.py

Lines changed: 25 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -941,17 +941,20 @@ class ISettings(SerializableMixin):
941941
cfgIntervalEnd: Optional[float] = None
942942
cfgIntervalStart: Optional[float] = None
943943
CFGScale: Optional[float] = None
944+
chunkLength: Optional[int] = None
944945
clipSkip: Optional[int] = None
945946
colorCorrection: Optional[bool] = None
946947
colorFix: Optional[bool] = None
947948
colorPalette: Optional[List[Union[IColorPaletteEntry, Dict[str, Any]]]] = None
948949
compress: Optional[str] = None
950+
conditionOnPreviousChunks: Optional[bool] = None
949951
controlNetWeight: Optional[float] = None
950952
coverConditioningScale: Optional[float] = None
951953
decimation: Optional[int] = None
952954
decimationTarget: Optional[int] = None
953955
draft: Optional[bool] = None
954956
editRegions: Optional[List[List[Union[IEditRegion, Dict[str, Any]]]]] = None
957+
earlyStopThreshold: Optional[float] = None
955958
emotion: Optional[str] = None
956959
enhanceDetails: Optional[bool] = None
957960
exportUv: Optional[bool] = None
@@ -974,6 +977,7 @@ class ISettings(SerializableMixin):
974977
keyScale: Optional[str] = None
975978
languageBoost: Optional[str] = None
976979
layers: Optional[int] = None
980+
latency: Optional[str] = None
977981
lyrics: Optional[str] = None
978982
lyricsOptimizer: Optional[bool] = None
979983
magicPrompt: Optional[str] = None
@@ -983,9 +987,12 @@ class ISettings(SerializableMixin):
983987
meshMode: Optional[str] = None
984988
meshType: Optional[str] = None
985989
minP: Optional[float] = None
990+
minChunkLength: Optional[int] = None
986991
mode: Optional[str] = None
987992
moderation: Optional[bool] = None
988993
multiClip: Optional[bool] = None
994+
normalize: Optional[bool] = None
995+
normalizeLoudness: Optional[bool] = None
989996
negativePrompt: Optional[str] = None
990997
occlusionDetection: Optional[bool] = None
991998
orientation: Optional[str] = None
@@ -2092,17 +2099,31 @@ class ITrainingResult:
20922099
outputs: Optional[Dict[str, Any]] = None
20932100

20942101

2102+
@dataclass
2103+
class IAudioReferenceVoice(SerializableMixin):
2104+
audio: Union[str, File]
2105+
text: str
2106+
2107+
20952108
@dataclass
20962109
class IAudioInputs(SerializableMixin):
20972110
audio: Optional[str] = None
20982111
audios: Optional[List[str]] = None
20992112
video: Optional[str] = None
21002113
videos: Optional[List[str]] = None
2114+
referenceVoices: Optional[List[Union[IAudioReferenceVoice, Dict[str, Any]]]] = None
21012115

21022116
@property
21032117
def request_key(self) -> str:
21042118
return "inputs"
21052119

2120+
def __post_init__(self) -> None:
2121+
if self.referenceVoices is not None:
2122+
self.referenceVoices = [
2123+
IAudioReferenceVoice(**ref) if isinstance(ref, dict) else ref
2124+
for ref in self.referenceVoices
2125+
]
2126+
21062127

21072128
@dataclass
21082129
class IAudioVoice(SerializableMixin):
@@ -2114,17 +2135,17 @@ class IAudioVoice(SerializableMixin):
21142135
class IAudioSpeech(SerializableMixin):
21152136
text: Optional[str] = None
21162137
voice: Optional[str] = None
2117-
voices: Optional[List[Union[IAudioVoice, Dict[str, Any]]]] = None
2138+
voices: Optional[List[Union[str, IAudioVoice, Dict[str, Any]]]] = None
21182139
language: Optional[str] = None
21192140
speed: Optional[float] = None
2120-
volume: Optional[int] = None
2141+
volume: Optional[float] = None
21212142
pitch: Optional[int] = None
21222143
emotion: Optional[str] = None
21232144
tone: Optional[List[str]] = None
21242145

2125-
def __post_init__(self):
2146+
def __post_init__(self) -> None:
21262147
if self.voices is not None and isinstance(self.voices, (list, tuple)):
2127-
normalized_voices = []
2148+
normalized_voices: List[Union[str, IAudioVoice, Dict[str, Any]]] = []
21282149
for v in self.voices:
21292150
if isinstance(v, dict):
21302151
normalized_voices.append(IAudioVoice(**v))

0 commit comments

Comments
 (0)