fix(api): pass inpaint param to checkpoint extraction

ssube · Jun 10, 2023 · e64b813 · e64b813
1 parent 665cf57
commit e64b813
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/api/onnx_web/convert/diffusion/checkpoint.py b/api/onnx_web/convert/diffusion/checkpoint.py
@@ -1709,6 +1709,7 @@ def convert_extract_checkpoint(
             dest,
             source,
             config_file=config_file,
+            is_inpainting=is_inpainting,
             vae_file=vae_file,
         ):
             logger.info("extracted checkpoint to Torch model: %s", working_name)