Added initial support for f-lite model

This commit is contained in:
Jaret Burkett
2025-05-01 11:15:18 -06:00
parent 5890e67a46
commit d9700bdb99
9 changed files with 1076 additions and 4 deletions

View File

@@ -2283,6 +2283,7 @@ class StableDiffusion:
bleed_latents: torch.FloatTensor = None,
is_input_scaled=False,
return_first_prediction=False,
bypass_guidance_embedding=False,
**kwargs,
):
timesteps_to_run = self.noise_scheduler.timesteps[start_timesteps:total_timesteps]
@@ -2299,6 +2300,7 @@ class StableDiffusion:
add_time_ids=add_time_ids,
is_input_scaled=is_input_scaled,
return_conditional_pred=True,
bypass_guidance_embedding=bypass_guidance_embedding,
**kwargs,
)
# some schedulers need to run separately, so do that. (euler for example)