[diffusion] refactor: remove hard-code of instanceof on PipelineConfig (#14186)
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
co-authored by
gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
parent
de153cf76a
commit
7ce8faae28
@@ -562,7 +562,6 @@ class VAELoader(ComponentLoader):
|
||||
|
||||
server_args.model_paths["vae"] = component_model_path
|
||||
|
||||
# TODO: abstract these logics
|
||||
logger.info("HF model config: %s", config)
|
||||
vae_config = server_args.pipeline_config.vae_config
|
||||
vae_config.update_model_arch(config)
|
||||
|
||||
Reference in New Issue
Block a user