ready for review

2024-04-12 11:36:52 +01:00
parent 5666ec3ec7
commit 6d0a45a97d
7 changed files with 11 additions and 42 deletions
--- a/lerobot/common/policies/diffusion/model/conditional_unet1d.py
+++ b/lerobot/common/policies/diffusion/model/conditional_unet1d.py
@@ -10,7 +10,6 @@ logger = logging.getLogger(__name__)


 class _SinusoidalPosEmb(nn.Module):
-    # TODO(now): consolidate?
    def __init__(self, dim):
        super().__init__()
        self.dim = dim