mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-19 16:02:11 +00:00
remove environment state conditioning
This commit is contained in:
@@ -755,9 +755,6 @@ class MultiTaskDiTPolicy(PreTrainedPolicy):
|
|||||||
if self.config.image_features:
|
if self.config.image_features:
|
||||||
self._queues["observation.images"] = deque(maxlen=self.config.n_obs_steps)
|
self._queues["observation.images"] = deque(maxlen=self.config.n_obs_steps)
|
||||||
|
|
||||||
if self.config.env_state_feature:
|
|
||||||
self._queues["observation.environment_state"] = deque(maxlen=self.config.n_obs_steps)
|
|
||||||
|
|
||||||
self._queues["task"] = deque(maxlen=self.config.n_obs_steps)
|
self._queues["task"] = deque(maxlen=self.config.n_obs_steps)
|
||||||
|
|
||||||
@torch.no_grad()
|
@torch.no_grad()
|
||||||
|
|||||||
Reference in New Issue
Block a user