• Rahul Raman's avatar
    example: Train Instruct pix2 pix with lora implementation (#6469) · 2d1f2182
    Rahul Raman authored
    
    
    * base template file - train_instruct_pix2pix.py
    
    * additional import and parser argument requried for lora
    
    * finetune only instructpix2pix model -- no need to include these layers
    
    * inject lora layers
    
    * freeze unet model -- only lora layers are trained
    
    * training modifications to train only lora parameters
    
    * store only lora parameters
    
    * move train script to research project
    
    * run quality and style code checks
    
    * move train script to a new folder
    
    * add README
    
    * update README
    
    * update references in README
    
    ---------
    Co-authored-by: default avatarRahul Raman <rahulraman@gmail.com>
    Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    2d1f2182
README.md 3.05 KB