Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

tidy(mm): "trigger_words" -> "trigger_phrases"

Browse Source
This commit is contained in:
psychedelicious
2024-03-05 11:10:18 +11:00
parent 93dd3c848e
commit c953e61294
5 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_manager/config.py
View File

@ -147,7 +147,7 @@ class ModelConfigBase(BaseModel):
source_api_response: Optional[str] = Field(
description="The original API response from the source, as stringified JSON.", default=None
)
trigger_words: Optional[set[str]] = Field(description="Set of trigger words for this model", default=None)
trigger_phrases: Optional[set[str]] = Field(description="Set of trigger phrases for this model", default=None)
model_config = ConfigDict(use_enum_values=False, validate_assignment=True)

6
invokeai/backend/model_manager/metadata/fetch/civitai.py
View File

@ -144,14 +144,14 @@ class CivitaiMetadataFetch(ModelMetadataFetchBase):
]
try:
trigger_words = StringSetAdapter.validate_python(version_json.get("trainedWords"))
trigger_phrases = StringSetAdapter.validate_python(version_json.get("trainedWords"))
except ValidationError:
trigger_words: set[str] = set()
trigger_phrases: set[str] = set()
return CivitaiMetadata(
name=version_json["name"],
files=model_files,
trigger_words=trigger_words,
trigger_phrases=trigger_phrases,
api_response=json.dumps(version_json),
)

2
invokeai/backend/model_manager/metadata/metadata_base.py
View File

@ -92,7 +92,7 @@ class CivitaiMetadata(ModelMetadataWithFiles):
"""Extended metadata fields provided by Civitai."""
type: Literal["civitai"] = "civitai"
trigger_words: set[str] = Field(description="Trigger words extracted from the API response")
trigger_phrases: set[str] = Field(description="Trigger phrases extracted from the API response")
api_response: Optional[str] = Field(description="Response from the Civitai API as stringified JSON", default=None)