• Henrik Forstén's avatar
    Controlnet training (#2545) · 79eb3d07
    Henrik Forstén authored
    * Controlnet training code initial commit
    
    Works with circle dataset: https://github.com/lllyasviel/ControlNet/blob/main/docs/train.md
    
    * Script for adding a controlnet to existing model
    
    * Fix control image transform
    
    Control image should be in 0..1 range.
    
    * Add license header and remove more unused configs
    
    * controlnet training readme
    
    * Allow nonlocal model in add_controlnet.py
    
    * Formatting
    
    * Remove unused code
    
    * Code quality
    
    * Initialize controlnet in training script
    
    * Formatting
    
    * Address review comments
    
    * doc style
    
    * explicit constructor args and submodule names
    
    * hub dataset
    
    NOTE -  not tested
    
    * empty prompts
    
    * add conditioning image
    
    * rename
    
    * remove instance data dir
    
    * image_transforms -> -1,1 . conditioning_image_transformers -> 0, 1
    
    * nits
    
    * remove local rank config
    
    I think this isn't necessary in any of our training scripts
    
    * validation images
    
    * proportion_empty_prompts typo
    
    * weight copy...
    79eb3d07
.gitignore 1.84 KB