Argument descriptions (#208)

* add argument descriptions

* update

* update

* update
This commit is contained in:
Aryan
2025-01-11 05:06:02 +05:30
committed by GitHub
parent b34091362e
commit 81bf590a91
3 changed files with 212 additions and 7 deletions
+1 -1
View File
@@ -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
+1 -1
View File
@@ -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,