huggingface / huggingface/diffusers
AutoencoderKLWan - support grandient_checkpointing
Open
contributions-welcome
stale
- Dominant language
- Python
- Stars
- 34.5k
- Forks
- 7.3k
- Avg merge
- 3d 3h
- Merged PRs (30d)
- 91
Description
Do you have a plan for supporting gradient checkpointing for AutoencoderKLWan?
Thank you for always working hard for open source :pray::pray:
Contributor guide
Research direction
Start by locating AutoencoderKLWan and reviewing how gradient checkpointing is supported in related components. Clarify the expected behavior and scope, then add support and tests that demonstrate checkpointing works for this autoencoder.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 30/100