-
Notifications
You must be signed in to change notification settings - Fork 5.8k
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
[Docs
] Update and make improvements
#5819
Conversation
The documentation is not available anymore as the PR was closed or merged. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Cool, thanks for optimizing!
I guess the type of timesteps should be strictly int64. The default type might be different from system to system. I should have posed the question why did someone put |
* Revert "[`Docs`] Update and make improvements (huggingface#5819)" This reverts commit c697f52. * Update README.md * Update memory.md * Update basic_training.md * Update write_own_pipeline.md * Update fp16.md * Update basic_training.md * Update write_own_pipeline.md * Update write_own_pipeline.md
* Revert "[`Docs`] Update and make improvements (huggingface#5819)" This reverts commit c697f52. * Update README.md * Update memory.md * Update basic_training.md * Update write_own_pipeline.md * Update fp16.md * Update basic_training.md * Update write_own_pipeline.md * Update write_own_pipeline.md
Update and make improvements
* Revert "[`Docs`] Update and make improvements (huggingface#5819)" This reverts commit c697f52. * Update README.md * Update memory.md * Update basic_training.md * Update write_own_pipeline.md * Update fp16.md * Update basic_training.md * Update write_own_pipeline.md * Update write_own_pipeline.md
What does this PR do?
Creating tensors directly on the GPU is ~22 times, and creating them directly as half-precision is ~4 times faster.
-- Experiment environment:
-- CPU: AMD Ryzen 5 3550H
-- GPU: NVIDIA GeForce GTX 1650 Mobile / Max-Q
-- Platform: Linux-6.2.0-36-generic-x86_64-with-glibc2.35
-- Python version: 3.10.12
-- PyTorch version: 2.1.1
Of course, at the end of the day, the time for creating tensors in the docs doesn't create a visible difference, but I edited them as best practices.
Removed .long() castings because the tensors were already created .long() format by default.
Before submitting
documentation guidelines, and
here are tips on formatting docstrings.
Who can review?
Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.
@stevhliu @yiyixuxu @DN6