Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix checkpoint load issue #11859

Merged
merged 4 commits into from
Jan 17, 2025
Merged

fix checkpoint load issue #11859

merged 4 commits into from
Jan 17, 2025

Conversation

dimapihtar
Copy link
Collaborator

What does this PR do ?

Add a one line overview of what this PR aims to accomplish.

Collection: [Note which collection this PR will affect]

Changelog

  • Add specific line by line info of high level changes in this PR.

Usage

  • You can potentially add a usage example below
# Add a code snippet demonstrating how to use this 

GitHub Actions CI

The Jenkins CI system has been replaced by GitHub Actions self-hosted runners.

The GitHub Actions CI will run automatically when the "Run CICD" label is added to the PR.
To re-run CI remove and add the label again.
To run CI on an untrusted fork, a NeMo user with write access must first click "Approve and run".

Before your PR is "Ready for review"

Pre checks:

  • Make sure you read and followed Contributor guidelines
  • Did you write any new necessary tests?
  • Did you add or update any necessary documentation?
  • Does the PR affect components that are optional to install? (Ex: Numba, Pynini, Apex etc)
    • Reviewer: Does the PR have correct import guards for all optional libraries?

PR Type:

  • New Feature
  • Bugfix
  • Documentation

If you haven't finished some of the above items you can still open "Draft" PR.

Who can review?

Anyone in the community is free to review the PR once the checks have passed.
Contributor guidelines contains specific people who can review PRs to various areas.

Additional Information

  • Related to # (issue)

Signed-off-by: Dmytro Pykhtar <dpykhtar@nvidia.com>
dimapihtar and others added 2 commits January 15, 2025 18:14
Signed-off-by: dimapihtar <dimapihtar@users.noreply.github.com>
Signed-off-by: dimapihtar <dpihtar@gmail.com>
@github-actions github-actions bot added the NLP label Jan 17, 2025
@dimapihtar dimapihtar marked this pull request as ready for review January 17, 2025 13:33
Copy link
Collaborator

@mikolajblaz mikolajblaz left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This MR doesn't change any behavior wrt. to old versions (in both new and old containers), so approving.

In the long run there is an open question of torch.load security.

Copy link
Contributor

beep boop 🤖: 🙏 The following files have warnings. In case you are familiar with these, please try helping us to improve the code base.


Your code was analyzed with PyLint. The following annotations have been identified:

************* Module nemo.collections.llm.gpt.model.ssm
nemo/collections/llm/gpt/model/ssm.py:190:0: C0301: Line too long (129/119) (line-too-long)
nemo/collections/llm/gpt/model/ssm.py:193:0: C0301: Line too long (131/119) (line-too-long)
nemo/collections/llm/gpt/model/ssm.py:194:0: C0301: Line too long (151/119) (line-too-long)
nemo/collections/llm/gpt/model/ssm.py:195:0: C0301: Line too long (129/119) (line-too-long)
nemo/collections/llm/gpt/model/ssm.py:39:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:51:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:89:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:110:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:127:8: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:131:12: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:134:12: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:155:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:203:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:216:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:221:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:235:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:249:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:263:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:277:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:291:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:306:0: C0115: Missing class docstring (missing-class-docstring)
************* Module nemo.collections.multimodal.models.text_to_image.controlnet.controlnet
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:148:0: C0301: Line too long (137/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:581:0: C0301: Line too long (125/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:597:0: C0301: Line too long (166/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:635:0: C0301: Line too long (149/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:973:0: C0301: Line too long (140/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:988:0: C0301: Line too long (155/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:1003:0: C0301: Line too long (137/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:1013:0: C0301: Line too long (125/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:110:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:164:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:269:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:282:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:544:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:605:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:608:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:631:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:660:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:673:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:685:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:811:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:858:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:865:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:944:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:1033:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py:1036:4: C0116: Missing function or method docstring (missing-function-docstring)
************* Module nemo.collections.multimodal.models.text_to_image.imagen.imagen_pipeline
nemo/collections/multimodal/models/text_to_image/imagen/imagen_pipeline.py:29:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/imagen/imagen_pipeline.py:39:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/imagen/imagen_pipeline.py:45:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/imagen/imagen_pipeline.py:61:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/imagen/imagen_pipeline.py:155:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/imagen/imagen_pipeline.py:183:4: C0116: Missing function or method docstring (missing-function-docstring)
************* Module nemo.collections.multimodal.models.text_to_image.instruct_pix2pix.ldm.ddpm_edit
nemo/collections/multimodal/models/text_to_image/instruct_pix2pix/ldm/ddpm_edit.py:126:0: C0301: Line too long (132/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/instruct_pix2pix/ldm/ddpm_edit.py:155:0: C0301: Line too long (123/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/instruct_pix2pix/ldm/ddpm_edit.py:205:0: C0301: Line too long (140/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/instruct_pix2pix/ldm/ddpm_edit.py:213:0: C0301: Line too long (155/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/instruct_pix2pix/ldm/ddpm_edit.py:225:0: C0301: Line too long (137/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/instruct_pix2pix/ldm/ddpm_edit.py:42:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/instruct_pix2pix/ldm/ddpm_edit.py:146:0: C0115: Missing class docstring (missing-class-docstring)
************* Module nemo.collections.multimodal.models.text_to_image.stable_diffusion.ldm.autoencoder
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:354:0: C0301: Line too long (214/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:33:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:76:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:90:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:104:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:108:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:114:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:119:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:124:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:129:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:136:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:154:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:184:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:227:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:254:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:257:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:280:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:289:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:310:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:478:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:506:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:515:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:522:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:557:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:573:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:586:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:590:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:606:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:615:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:620:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:625:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:628:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:631:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/autoencoder.py:636:4: C0116: Missing function or method docstring (missing-function-docstring)
************* Module nemo.collections.multimodal.models.text_to_image.stable_diffusion.ldm.ddpm
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1109:0: C0301: Line too long (141/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1678:0: C0301: Line too long (149/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:2072:0: C0301: Line too long (140/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:2096:0: C0301: Line too long (155/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:2111:0: C0301: Line too long (137/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:2121:0: C0301: Line too long (125/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:2205:0: C0301: Line too long (127/119) (line-too-long)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:120:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:124:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:172:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:240:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:324:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:330:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:336:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:342:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:351:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:364:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:373:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:389:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:396:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:403:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:409:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:424:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:455:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:461:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:472:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:485:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:581:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:588:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:615:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:622:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:654:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:663:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:676:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:697:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:851:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:913:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:975:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1039:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1303:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1498:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1647:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1658:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1716:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1729:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1741:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1865:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1920:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1969:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:1975:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:2037:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:2141:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:2357:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:2378:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:18:0: W0611: Unused Dict imported from typing (unused-import)
nemo/collections/multimodal/models/text_to_image/stable_diffusion/ldm/ddpm.py:73:0: W0611: Unused get_last_rank imported from nemo.collections.nlp.parts.utils_funcs (unused-import)
************* Module nemo.collections.multimodal.modules.stable_diffusion.diffusionmodules.openaimodel
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:559:0: C0301: Line too long (120/119) (line-too-long)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:562:0: C0301: Line too long (128/119) (line-too-long)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:566:0: C0301: Line too long (120/119) (line-too-long)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:968:0: C0301: Line too long (187/119) (line-too-long)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:1245:0: C0301: Line too long (121/119) (line-too-long)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:1344:0: C0301: Line too long (120/119) (line-too-long)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:56:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:70:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:74:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:78:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:125:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:189:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:221:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:253:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:432:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:490:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:531:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:535:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:540:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:1152:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:1381:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py:20:0: W0611: Unused partial imported from functools (unused-import)
************* Module nemo.collections.multimodal.speech_llm.models.modular_models
nemo/collections/multimodal/speech_llm/models/modular_models.py:424:0: C0301: Line too long (121/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:459:0: C0301: Line too long (132/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:715:0: C0301: Line too long (140/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:786:0: C0301: Line too long (127/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:787:0: C0301: Line too long (168/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1151:0: C0301: Line too long (139/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1186:0: C0301: Line too long (254/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1220:0: C0301: Line too long (125/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1225:0: C0301: Line too long (135/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1234:0: C0301: Line too long (124/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1241:0: C0301: Line too long (140/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1245:0: C0301: Line too long (159/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1301:0: C0301: Line too long (122/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1347:0: C0301: Line too long (145/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1350:0: C0301: Line too long (145/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1447:0: C0301: Line too long (130/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1466:0: C0301: Line too long (124/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1471:0: C0301: Line too long (146/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1550:0: C0301: Line too long (135/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1695:0: C0301: Line too long (142/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1696:0: C0301: Line too long (145/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1732:0: C0301: Line too long (128/119) (line-too-long)
nemo/collections/multimodal/speech_llm/models/modular_models.py:97:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/speech_llm/models/modular_models.py:869:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/speech_llm/models/modular_models.py:893:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1072:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/multimodal/speech_llm/models/modular_models.py:1636:4: C0116: Missing function or method docstring (missing-function-docstring)
************* Module nemo.collections.multimodal.speech_llm.parts.mixins.adapter_mixin
nemo/collections/multimodal/speech_llm/parts/mixins/adapter_mixin.py:73:0: C0301: Line too long (166/119) (line-too-long)
nemo/collections/multimodal/speech_llm/parts/mixins/adapter_mixin.py:25:0: C0115: Missing class docstring (missing-class-docstring)
************* Module nemo.collections.nlp.parts.mixins.multimodal_adapter_mixins
nemo/collections/nlp/parts/mixins/multimodal_adapter_mixins.py:34:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/nlp/parts/mixins/multimodal_adapter_mixins.py:26:4: W0611: Unused parallel_state imported from megatron.core (unused-import)
************* Module nemo.collections.nlp.parts.mixins.nlp_adapter_mixins
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:69:0: C0301: Line too long (139/119) (line-too-long)
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:75:0: C0301: Line too long (139/119) (line-too-long)
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:86:0: C0301: Line too long (124/119) (line-too-long)
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:130:0: C0301: Line too long (131/119) (line-too-long)
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:432:0: C0301: Line too long (123/119) (line-too-long)
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:448:0: C0301: Line too long (170/119) (line-too-long)
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:455:0: C0301: Line too long (139/119) (line-too-long)
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:59:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:114:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:388:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:396:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:446:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:458:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/nlp/parts/mixins/nlp_adapter_mixins.py:465:4: C0116: Missing function or method docstring (missing-function-docstring)

-----------------------------------
Your code has been rated at 9.56/10

Mitigation guide:

  • Add sensible and useful docstrings to functions and methods
  • For trivial methods like getter/setters, consider adding # pylint: disable=C0116 inside the function itself
  • To disable multiple functions/methods at once, put a # pylint: disable=C0116 before the first and a # pylint: enable=C0116 after the last.

By applying these rules, we reduce the occurance of this message in future.

Thank you for improving NeMo's documentation!

@dimapihtar dimapihtar enabled auto-merge (squash) January 17, 2025 13:39
Copy link
Contributor

[🤖]: Hi @dimapihtar 👋,

We wanted to let you know that a CICD pipeline for this PR just finished successfully

So it might be time to merge this PR or get some approvals

I'm just a bot so I'll leave it you what to do next.

//cc @pablo-garay @ko3n1g

@dimapihtar dimapihtar merged commit 0cd990d into main Jan 17, 2025
208 of 212 checks passed
@dimapihtar dimapihtar deleted the dpykhtar/pyt_tot branch January 17, 2025 18:04
abhinavg4 pushed a commit that referenced this pull request Jan 30, 2025
* fix checkpoint load issue

Signed-off-by: Dmytro Pykhtar <dpykhtar@nvidia.com>

* Apply isort and black reformatting

Signed-off-by: dimapihtar <dimapihtar@users.noreply.github.com>

* set weights_only to False

Signed-off-by: dimapihtar <dpihtar@gmail.com>

---------

Signed-off-by: Dmytro Pykhtar <dpykhtar@nvidia.com>
Signed-off-by: dimapihtar <dimapihtar@users.noreply.github.com>
Signed-off-by: dimapihtar <dpihtar@gmail.com>
Co-authored-by: dimapihtar <dimapihtar@users.noreply.github.com>
Signed-off-by: Abhinav Garg <abhgarg@nvidia.com>
youngeunkwon0405 pushed a commit to youngeunkwon0405/NeMo that referenced this pull request Feb 10, 2025
* fix checkpoint load issue

Signed-off-by: Dmytro Pykhtar <dpykhtar@nvidia.com>

* Apply isort and black reformatting

Signed-off-by: dimapihtar <dimapihtar@users.noreply.github.com>

* set weights_only to False

Signed-off-by: dimapihtar <dpihtar@gmail.com>

---------

Signed-off-by: Dmytro Pykhtar <dpykhtar@nvidia.com>
Signed-off-by: dimapihtar <dimapihtar@users.noreply.github.com>
Signed-off-by: dimapihtar <dpihtar@gmail.com>
Co-authored-by: dimapihtar <dimapihtar@users.noreply.github.com>
Signed-off-by: Youngeun Kwon <youngeunk@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants