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

Display class name in model error message

Browse Source 
Signed-off-by: Emmanuel Ferdman <emmanuelferdman@gmail.com>
This commit is contained in:
Emmanuel Ferdman
2025-04-25 09:52:25 -07:00
parent 676af4c4f0
commit 0d34b3df6d
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."
) )