naifu
Train generative models with pytorch lightning
File Explorer
Download Latest Version (.zip)- __init__.py
- deepspeed.py
- fairscale.py
- fsdp.py
- logging.py
- smddp.py
- trainer.py
- utils.py
- train_cascade_stage_c.yaml
- train_diffusers.yaml
- train_dpo.yaml
- train_dpo_diffusers.yaml
- train_dpo_sd15_diffusers.yaml
- train_fairscale.yaml
- train_general_llm.yaml
- train_gpt2.yaml
- train_lcm.yaml
- train_llava.yaml
- train_llava_1.6.yaml
- train_lycoris.yaml
- train_lycoris_scaled.yaml
- train_mistral_lora.yaml
- train_phi2.yaml
- train_pixart.yaml
- train_pixart_sigma.yaml
- train_refiner.yaml
- train_sd15.yaml
- train_sdxl.yaml
- train_sdxl_adaptive.yaml
- train_sdxl_deepspeed.yaml
- train_sdxl_original.yaml
- train_sdxl_v.yaml
- __init__.py
- bucket.py
- conversation.py
- image_storage.py
- llava_dataset.py
- paired_wds.py
- processors.py
- prompt_style.py
- text_dataset.py
- common.py
- effnet.py
- previewer.py
- quantize.py
- stage_a.py
- stage_b.py
- stage_c.py
- __init__.py
- loss_weights.py
- noise_conditions.py
- readme.md
- samplers.py
- scalers.py
- schedulers.py
- targets.py
- axial_rope.py
- dctorch.py
- flags.py
- flops.py
- layers.py
- network.py
- utils.py
- constants.py
- llava_llama.py
- mm_utils.py
- configuration_phi.py
- modeling_opt.py
- modeling_phi.py
- __init__.py
- alpha.py
- diffusion.py
- sigma.py
- __init__.py
- attention.py
- autoencoder.py
- denoiser.py
- denoiser_scaling.py
- denoiser_weighting.py
- discretizer.py
- encoder_util.py
- encoders.py
- model.py
- model_util.py
- sigma_sampling.py
- vae_model.py
- __init__.py
- __init__.py
- cascade_model.py
- config_sdxl_base.py
- config_sdxl_refiner.py
- scheduler_utils.py
- sd_dpo_diffusers.py
- sd_model_diffusers.py
- sdxl_dpo.py
- sdxl_dpo_diffusers.py
- sdxl_model.py
- sdxl_model_diffusers.py
- sdxl_utils.py
- train_cascade_stage_c.py
- train_diffusers.py
- train_general_llm.py
- train_gpt2.py
- train_lcm.py
- train_llava.py
- train_lycoris.py
- train_phi.py
- train_pixart.py
- train_pixart_sigma.py
- train_refiner.py
- train_sd_diffusers.py
- train_sdxl.py
- train_sdxl_edm.py
- train_sdxl_flow.py
- train_sdxl_original.py
- deepdanbooru.py
- encode_latents_xl.py
- gpt2_inference.py
- run_llava.py
- vae_cache_inspector.ipynb
- vae_playground.ipynb
- wd14_tagger.py
- .gitignore
- LICENSE
- README.md
- requirements.txt
- trainer.py
// repository documentation
Was this content helpful?
(0 ratings)