• Gæros's avatar
    [LoRA] text encoder: read the ranks for all the attn modules (#8324) · 298ce679
    Gæros authored
    
    
    * [LoRA] text encoder: read the ranks for all the attn modules
    
     * In addition to out_proj, read the ranks of adapters for q_proj, k_proj, and  v_proj
    
     * Allow missing adapters (UNet already supports this)
    
    * ruff format loaders.lora
    
    * [LoRA] add tests for partial text encoders LoRAs
    
    * [LoRA] update test_simple_inference_with_partial_text_lora to be deterministic
    
    * [LoRA] comment justifying test_simple_inference_with_partial_text_lora
    
    * style
    
    ---------
    Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    298ce679
utils.py 64.5 KB