mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-03-13 06:29:48 +00:00
Added initial support for f-lite model
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user