[fix]:flux2 i2i auto target shape

[helloyongyang]

No description provided.

[gemini-code-assist]

Code Review

This pull request refactors the target shape assignment logic in the Flux2Klein runner, moving the i2i task shape initialization to the input encoder. However, the changes introduce a NameError in set_target_shape by using multiple_of before its definition. Additionally, the target_shape is not correctly updated when a custom shape is provided, and useful logging was removed.

[gemini-code-assist]

The set_target_shape method contains a critical bug and a logic error:

1. NameError: multiple_of is used on lines 265 and 266 before it is defined on line 269. This will cause a crash during execution for the t2i task.
2. Logic Error: self.input_info.target_shape is only updated inside the else block (line 267). If a custom_shape is provided (line 261), the target_shape attribute in input_info is never updated, which will lead to incorrect behavior in subsequent pipeline stages that rely on this value.
3. Missing Logging: The useful log message indicating the final target shape was removed.

I recommend moving the multiple_of definition to the top and ensuring target_shape is always assigned for the t2i task.

Suggested change

	task = self.config.get("task", "t2i")
	if task == "i2i" and hasattr(self.input_info, "auto_width"):
	width = self.input_info.auto_width
	height = self.input_info.auto_height
	else:
	if task == "i2i": # for i2i task, the target shape is already set in _run_input_encoder_local_i2i
	height, width = self.input_info.target_shape
	else: # for t2i task, calculate the target shape based on the resolution
	custom_shape = self.get_custom_shape()
	if custom_shape is not None:
	width, height = custom_shape
	else:
	calculated_width, calculated_height, _ = calculate_dimensions(self.resolution * self.resolution, 16 / 9)
	width = calculated_width // multiple_of * multiple_of
	height = calculated_height // multiple_of * multiple_of
	self.input_info.auto_width = width
	self.input_info.auto_height = height
	self.input_info.target_shape = (height, width)
	logger.info(f"Flux2Klein Image Runner set target shape: {width}x{height}")
	self.input_info.target_shape = (height, width)
	multiple_of = self.config.get("vae_scale_factor", 8) * 2
	multiple_of = self.config.get("vae_scale_factor", 8) * 2
	task = self.config.get("task", "t2i")
	if task == "i2i": # for i2i task, the target shape is already set in _run_input_encoder_local_i2i
	height, width = self.input_info.target_shape
	else: # for t2i task, calculate the target shape based on the resolution
	custom_shape = self.get_custom_shape()
	if custom_shape is not None:
	width, height = custom_shape
	else:
	calculated_width, calculated_height, _ = calculate_dimensions(self.resolution * self.resolution, 16 / 9)
	width = calculated_width // multiple_of * multiple_of
	height = calculated_height // multiple_of * multiple_of
	self.input_info.target_shape = (height, width)
	logger.info(f"Flux2Klein Image Runner set target shape: {width}x{height}")