fix: 修复单帧图生视频 model_key 转换逻辑
Browse files修复 I2V 单帧模式下 _fl 后缀去除失败的问题。
原逻辑仅处理 _fl_ 在中间的情况,但实际模型名称 _fl 在结尾:
- veo_3_1_i2v_s_fast_ultra_fl 应转为 veo_3_1_i2v_s_fast_ultra
现在同时处理两种情况:
1. _fl_ 在中间: replace("_fl_", "_")
2. _fl 在结尾: endswith("_fl") 后截取
Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
src/services/generation_handler.py
CHANGED
|
@@ -1121,10 +1121,12 @@ class GenerationHandler:
|
|
| 1121 |
user_paygate_tier=token.user_paygate_tier or "PAYGATE_TIER_ONE"
|
| 1122 |
)
|
| 1123 |
else:
|
| 1124 |
-
# 只有首帧 - 需要
|
| 1125 |
-
#
|
| 1126 |
-
#
|
| 1127 |
actual_model_key = model_config["model_key"].replace("_fl_", "_")
|
|
|
|
|
|
|
| 1128 |
debug_logger.log_info(f"[I2V] 单帧模式,model_key: {model_config['model_key']} -> {actual_model_key}")
|
| 1129 |
result = await self.flow_client.generate_video_start_image(
|
| 1130 |
at=token.at,
|
|
|
|
| 1121 |
user_paygate_tier=token.user_paygate_tier or "PAYGATE_TIER_ONE"
|
| 1122 |
)
|
| 1123 |
else:
|
| 1124 |
+
# 只有首帧 - 需要去掉 model_key 中的 _fl
|
| 1125 |
+
# 情况1: _fl_ 在中间 (如 veo_3_1_i2v_s_fast_fl_ultra_relaxed -> veo_3_1_i2v_s_fast_ultra_relaxed)
|
| 1126 |
+
# 情况2: _fl 在结尾 (如 veo_3_1_i2v_s_fast_ultra_fl -> veo_3_1_i2v_s_fast_ultra)
|
| 1127 |
actual_model_key = model_config["model_key"].replace("_fl_", "_")
|
| 1128 |
+
if actual_model_key.endswith("_fl"):
|
| 1129 |
+
actual_model_key = actual_model_key[:-3]
|
| 1130 |
debug_logger.log_info(f"[I2V] 单帧模式,model_key: {model_config['model_key']} -> {actual_model_key}")
|
| 1131 |
result = await self.flow_client.generate_video_start_image(
|
| 1132 |
at=token.at,
|