EveryDream2trainer/utils
Victor Hall fb1caf52a7 add llava captioning 2024-05-04 22:24:24 -04: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 patch cog to fix issue with transformers 2024-03-15 00:18:55 -04:00
sample_generator.py Merge branch 'victorchall:main' into feat_add_sde_samplers 2023-10-23 00:04:03 +02: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