let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #638 from emmanuel-ferdman/main

Browse Source 
Display class name in model error message
This commit is contained in:
Qing
2025-04-29 10:13:17 +08:00
committed by GitHub
parent 676af4c4f0 0d34b3df6d
commit 61a759fb3f
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
iopaint/model/brushnet/pipeline_brushnet.py
View File

@@ -201,7 +201,7 @@ class StableDiffusionBrushNetPipeline(
if safety_checker is not None and feature_extractor is None: if safety_checker is not None and feature_extractor is None:
raise ValueError( raise ValueError(
"Make sure to define a feature extractor when loading {self.__class__} if you want to use the safety" f"Make sure to define a feature extractor when loading {self.__class__} if you want to use the safety"
" checker. If you do not want to use the safety checker, you can pass `'safety_checker=None'` instead." " checker. If you do not want to use the safety checker, you can pass `'safety_checker=None'` instead."
) )

2
iopaint/model/power_paint/pipeline_powerpaint.py
View File

@@ -286,7 +286,7 @@ class StableDiffusionInpaintPipeline(
if safety_checker is not None and feature_extractor is None: if safety_checker is not None and feature_extractor is None:
raise ValueError( raise ValueError(
"Make sure to define a feature extractor when loading {self.__class__} if you want to use the safety" f"Make sure to define a feature extractor when loading {self.__class__} if you want to use the safety"
" checker. If you do not want to use the safety checker, you can pass `'safety_checker=None'` instead." " checker. If you do not want to use the safety checker, you can pass `'safety_checker=None'` instead."
) )

2
iopaint/model/power_paint/v2/pipeline_PowerPaint_Brushnet_CA.py
View File

@@ -222,7 +222,7 @@ class StableDiffusionPowerPaintBrushNetPipeline(
if safety_checker is not None and feature_extractor is None: if safety_checker is not None and feature_extractor is None:
raise ValueError( raise ValueError(
"Make sure to define a feature extractor when loading {self.__class__} if you want to use the safety" f"Make sure to define a feature extractor when loading {self.__class__} if you want to use the safety"
" checker. If you do not want to use the safety checker, you can pass `'safety_checker=None'` instead." " checker. If you do not want to use the safety checker, you can pass `'safety_checker=None'` instead."
) )