"docs/source/en/vscode:/vscode.git/clone" did not exist on "b36cbd4fba877a378bd68bb5900e0bb6d99076a8"
  • Ruoxi's avatar
    Implement `CustomDiffusionAttnProcessor2_0`. (#4604) · 16b9a57d
    Ruoxi authored
    * Implement `CustomDiffusionAttnProcessor2_0`
    
    * Doc-strings and type annotations for `CustomDiffusionAttnProcessor2_0`. (#1)
    
    * Update attnprocessor.md
    
    * Update attention_processor.py
    
    * Interops for `CustomDiffusionAttnProcessor2_0`.
    
    * Formatted `attention_processor.py`.
    
    * Formatted doc-string in `attention_processor.py`
    
    * Conditional CustomDiffusion2_0 for training example.
    
    * Remove unnecessary reference impl in comments.
    
    * Fix `save_attn_procs`.
    16b9a57d
train_custom_diffusion.py 57.3 KB