[diffusion] fix: fix Flux.2 condition image resize (#14232)
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
03888b9de5
commit
3ab8ae6847
@@ -260,12 +260,15 @@ def get_model_info(model_path: str) -> Optional[ModelInfo]:
|
||||
)
|
||||
return None
|
||||
|
||||
# 4. Combine and return the complete model info
|
||||
return ModelInfo(
|
||||
# 4. Combine the complete model info
|
||||
model_info = ModelInfo(
|
||||
pipeline_cls=pipeline_cls,
|
||||
sampling_param_cls=config_info.sampling_param_cls,
|
||||
pipeline_config_cls=config_info.pipeline_config_cls,
|
||||
)
|
||||
logger.info(f"Found model info: {model_info}")
|
||||
|
||||
return model_info
|
||||
|
||||
|
||||
# Registration of model configs
|
||||
|
||||
Reference in New Issue
Block a user