diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index f48259a0..b127c901 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -40,7 +40,7 @@ repos: - horde_safety==0.2.3 - torch==2.3.1 - ruamel.yaml - - horde_engine==2.15.2 - - horde_sdk==0.14.10 + - horde_engine==2.15.3 + - horde_sdk==0.14.11 - horde_model_reference==0.9.0 - semver diff --git a/horde-bridge.cmd b/horde-bridge.cmd index fb9ac750..adce7be4 100644 --- a/horde-bridge.cmd +++ b/horde-bridge.cmd @@ -5,7 +5,7 @@ cd /d %~dp0 call runtime python -s -m pip -V call python -s -m pip uninstall hordelib -call python -s -m pip install horde_sdk~=0.14.10 horde_model_reference~=0.9.0 horde_engine~=2.15.2 horde_safety~=0.2.3 -U +call python -s -m pip install horde_sdk~=0.14.11 horde_model_reference~=0.9.0 horde_engine~=2.15.3 horde_safety~=0.2.3 -U if %ERRORLEVEL% NEQ 0 ( echo "Please run update-runtime.cmd." diff --git a/horde_worker_regen/__init__.py b/horde_worker_regen/__init__.py index 7e5da517..8f6cd7d9 100644 --- a/horde_worker_regen/__init__.py +++ b/horde_worker_regen/__init__.py @@ -8,7 +8,7 @@ ASSETS_FOLDER_PATH = Path(__file__).parent / "assets" -__version__ = "9.0.4" +__version__ = "9.0.5" import pkg_resources # noqa: E402 diff --git a/horde_worker_regen/_version_meta.json b/horde_worker_regen/_version_meta.json index 22a5be22..839fa614 100644 --- a/horde_worker_regen/_version_meta.json +++ b/horde_worker_regen/_version_meta.json @@ -1,5 +1,5 @@ { - "recommended_version": "9.0.4", + "recommended_version": "9.0.5", "required_min_version": "9.0.2", "required_min_version_update_date": "2024-09-26", "required_min_version_info": { diff --git a/horde_worker_regen/process_management/process_manager.py b/horde_worker_regen/process_management/process_manager.py index 78043b4d..8d63cdd4 100644 --- a/horde_worker_regen/process_management/process_manager.py +++ b/horde_worker_regen/process_management/process_manager.py @@ -1816,12 +1816,17 @@ def receive_and_handle_process_messages(self) -> None: completed_job_info.state = GENERATION_STATE.csam elif message.safety_evaluations[i].is_nsfw: new_meta_entry = GenMetadataEntry( - type=METADATA_TYPE.censorship, + type=METADATA_TYPE.information, value=METADATA_VALUE.nsfw, ) - completed_job_info.job_image_results[i].generation_faults.append(new_meta_entry) - if completed_job_info.state != GENERATION_STATE.csam: - completed_job_info.state = GENERATION_STATE.censored + if message.safety_evaluations[i].replacement_image_base64 is not None: + new_meta_entry = GenMetadataEntry( + type=METADATA_TYPE.censorship, + value=METADATA_VALUE.nsfw, + ) + completed_job_info.job_image_results[i].generation_faults.append(new_meta_entry) + if completed_job_info.state != GENERATION_STATE.csam: + completed_job_info.state = GENERATION_STATE.censored else: completed_job_info.censored = False # logger.debug([c.generation_faults for c in completed_job_info.job_image_results]) diff --git a/pyproject.toml b/pyproject.toml index 257b310e..4dcd9376 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "horde_worker_regen" -version = "9.0.4" +version = "9.0.5" description = "Allows you to connect to the AI Horde and generate images for users." authors = [ {name = "tazlin", email = "tazlin.on.github@gmail.com"}, diff --git a/requirements.rocm.txt b/requirements.rocm.txt index 5208c77b..f7dc6864 100644 --- a/requirements.rocm.txt +++ b/requirements.rocm.txt @@ -1,9 +1,9 @@ numpy==1.26.4 torch==2.3.1+rocm6.0 -horde_sdk~=0.14.10 +horde_sdk~=0.14.11 horde_safety~=0.2.3 -horde_engine~=2.15.2 +horde_engine~=2.15.3 horde_model_reference~=0.9.0 python-dotenv diff --git a/requirements.txt b/requirements.txt index 8c29d044..afd6eb0b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,9 +1,9 @@ numpy==1.26.4 torch==2.3.1 -horde_sdk~=0.14.10 +horde_sdk~=0.14.11 horde_safety~=0.2.3 -horde_engine~=2.15.2 +horde_engine~=2.15.3 horde_model_reference>=0.9.0 python-dotenv