• Chi's avatar
    Replacing the nn.Mish activation function with a get_activation function. (#5651) · 9ae90593
    Chi authored
    
    
    * I added a new doc string to the class. This is more flexible to understanding other developers what are doing and where it's using.
    
    * Update src/diffusers/models/unet_2d_blocks.py
    
    This changes suggest by maintener.
    Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    
    * Update src/diffusers/models/unet_2d_blocks.py
    
    Add suggested text
    Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    
    * Update unet_2d_blocks.py
    
    I changed the Parameter to Args text.
    
    * Update unet_2d_blocks.py
    
    proper indentation set in this file.
    
    * Update unet_2d_blocks.py
    
    a little bit of change in the act_fun argument line.
    
    * I run the black command to reformat style in the code
    
    * Update unet_2d_blocks.py
    
    similar doc-string add to have in the original diffusion repository.
    
    * I removed the dummy variable defined in both the encoder and decoder.
    
    * Now, I run black package to reformat my file
    
    * Remove the redundant line from the adapter.py file.
    
    * Black package using to reformated my file
    
    * Replacing the nn.Mish activation function with a get_activation function allows developers to more easily choose the right activation function for their task. Additionally, removing redundant variables can improve code readability and maintainability.
    
    * I try to fix this: Fast tests for PRs / Fast PyTorch Models & Schedulers CPU tests (pull_request)
    
    * Update src/diffusers/models/resnet.py
    Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
    
    ---------
    Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    Co-authored-by: default avatarDhruv Nair <dhruv.nair@gmail.com>
    Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
    9ae90593
vq_model.py 7.06 KB