mirror of
https://github.com/storytold/FineTrainers-Conditioning.git
synced 2026-10-18 07:33:20 +00:00
Argument descriptions (#208)
* add argument descriptions * update * update * update
This commit is contained in:
@@ -5,7 +5,7 @@ from .diffusion_utils import (
|
||||
prepare_loss_weights,
|
||||
prepare_sigmas,
|
||||
prepare_target,
|
||||
resolution_dependant_timestep_flow_shift,
|
||||
resolution_dependent_timestep_flow_shift,
|
||||
)
|
||||
from .file_utils import delete_files, find_files
|
||||
from .memory_utils import bytes_to_gigabytes, free_memory, get_memory_statistics, make_contiguous
|
||||
|
||||
@@ -7,7 +7,7 @@ from diffusers.training_utils import compute_loss_weighting_for_sd3
|
||||
|
||||
|
||||
# Default values copied from https://github.com/huggingface/diffusers/blob/8957324363d8b239d82db4909fbf8c0875683e3d/src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py#L47
|
||||
def resolution_dependant_timestep_flow_shift(
|
||||
def resolution_dependent_timestep_flow_shift(
|
||||
latents: torch.Tensor,
|
||||
sigmas: torch.Tensor,
|
||||
base_image_seq_len: int = 256,
|
||||
|
||||
Reference in New Issue
Block a user