scottshireman dc756ce22a Update sample_generator.py ... Changed {sample_index} to {sample_index:03} in file names and logs for sample generator to allow for proper sorting in wandb and other places when there are more than 10 samples as currently wandb sorts as follows: 0, 1, 10, 11, 2, etc. With this change it would be 000, 001, 002, ... , 010, 011, etc.		2024-12-03 08:47:28 -05:00
..
backdate_vae_keys.py	fix reshaping for picking apps	2023-11-29 22:17:52 -05:00
check_git.py	more wip optimizer splitting	2023-05-04 20:15:36 -04:00
convert_diff_to_ckpt.py	save as safetensors not ckpt	2023-11-13 13:12:42 -05:00
convert_diffusers_to_stable_diffusion.py	issue with conv	2023-01-12 20:59:04 -05:00
convert_original_stable_diffusion_to_diffusers.py	fix ckpt to diffusers conversion script	2023-11-15 12:55:22 -05:00
ed_logging.py	add llava captioning	2024-05-04 22:24:24 -04:00
first_fit_decreasing.py	remove deepcopies	2023-06-08 20:34:51 +02:00
fs_helpers.py	add huber loss, timestep clamping, slightly safer txt reading	2024-04-26 23:54:31 -04:00
general.py	fix ckpt to diffusers conversion script	2023-11-15 12:55:22 -05:00
get_yamls.py	bunch of updates, grad ckpting, no drop bucket, shuffle every epoch	2023-01-01 10:45:18 -05:00
gpu.py	fix mem leak on huge data, rework optimizer to separate json, add lion optimizer	2023-02-25 15:05:22 -05:00
huggingface_downloader.py	remove errant input	2023-07-08 14:59:38 -04:00
isolate_rng.py	workaround for cuda errors on some rented machines	2023-11-15 14:45:43 -05:00
log_wrapper.py	merge	2023-01-15 22:07:37 -05:00
patch_bnb.py	hey look ed2	2022-12-17 22:32:48 -05:00
patch_cog.py	add kwargs to cog patch for compatibility with newer diffusers versions	2024-06-18 18:24:23 -04:00
sample_generator.py	Update sample_generator.py	2024-12-03 08:47:28 -05:00
split_dataset.py	some validation tweaks and fixes	2023-02-19 10:20:16 +01:00
unet_utils.py	remove failure when people do weird custom configs on models	2023-06-10 16:29:03 -04:00