refine training

This commit is contained in:
Artiprocher
2025-12-01 15:08:45 +08:00
parent 9048d2e9d4
commit 62c94a9927
5 changed files with 10 additions and 8 deletions

View File

@@ -89,7 +89,7 @@ class FlowMatchScheduler():
return float(mu)
@staticmethod
def set_timesteps_flux2(num_inference_steps=100, denoising_strength=1.0, dynamic_shift_len=None):
def set_timesteps_flux2(num_inference_steps=100, denoising_strength=1.0, dynamic_shift_len=1024//16*1024//16):
sigma_min = 1 / num_inference_steps
sigma_max = 1.0
num_train_timesteps = 1000