admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-21 03:30:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

removing dataclass

Browse Source
This commit is contained in:
Maximellerbach
2026-03-27 16:53:45 +01:00
committed by Steven Palma
parent 0b957bb4b8
commit 68173f05df
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/groot/action_head/flow_matching_action_head.py
+1 -2
View File
@@ -13,7 +13,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
from dataclasses import dataclass, field from dataclasses import field
from typing import TYPE_CHECKING from typing import TYPE_CHECKING
import torch import torch
@@ -109,7 +109,6 @@ class MultiEmbodimentActionEncoder(nn.Module):
return x return x
@dataclass
class FlowmatchingActionHeadConfig(PretrainedConfig): class FlowmatchingActionHeadConfig(PretrainedConfig):
"""NOTE: N1.5 uses XEmbFlowmatchingPolicyHeadConfig as action head""" """NOTE: N1.5 uses XEmbFlowmatchingPolicyHeadConfig as action head"""