Skip to content

Commit 25c63cc

Browse files
mwulfmanMathias Wulfman
and
Mathias Wulfman
authored
🐛 Remove map_location=device that no longer exists when loading DiffusionPolicy from_pretained after commit 5e94738 (#830)
Co-authored-by: Mathias Wulfman <mathias.wulfman@wandercraft.eu>
1 parent 5e94738 commit 25c63cc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/2_evaluate_pretrained_policy.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444
# OR a path to a local outputs/train folder.
4545
# pretrained_policy_path = Path("outputs/train/example_pusht_diffusion")
4646

47-
policy = DiffusionPolicy.from_pretrained(pretrained_policy_path, map_location=device)
47+
policy = DiffusionPolicy.from_pretrained(pretrained_policy_path)
4848

4949
# Initialize evaluation environment to render two observation types:
5050
# an image of the scene and state/position of the agent. The environment

0 commit comments

Comments
 (0)