icrews Claude Sonnet 4.5 commited on
Commit
a570ba7
·
1 Parent(s): 6e0803d

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>

Files changed (1) hide show
  1. src/services/generation_handler.py +5 -3
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
- # 只有首帧 - 需要 model_key 中的 _fl_ 替换为 _
1125
- # 例如: veo_3_1_i2v_s_fast_fl_ultra_relaxed -> veo_3_1_i2v_s_fast_ultra_relaxed
1126
- # veo_3_1_i2v_s_fast_portrait_fl_ultra_relaxed -> veo_3_1_i2v_s_fast_portrait_ultra_relaxed
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,