-
Notifications
You must be signed in to change notification settings - Fork 34
Open
Description
In diffusion/model/nets/pixart_relactrl_v1.py, the code uses DropPath class but doesn't import it directly. The class is referenced in the RGLC_Block constructor but there's no corresponding import statement.
Steps to Reproduce
Check the code at line ~100 in diffusion/model/nets/pixart_relactrl_v1.py:
self.drop_path = DropPath(drop_path) if drop_path > 0. else nn.Identity()Proposed Solution
Add the following import to the top of the file:
from timm.models.layers import DropPathThis is consistent with other model files in the repository that properly import this class.
Nice works!
Metadata
Metadata
Assignees
Labels
No labels