 Pepijn
|
64974c38c2
|
cleanup tests
|
2025-09-17 17:35:07 +02:00 |
|
|
Pepijn
|
bc10fc7696
|
fix doc and constants
|
2025-09-17 17:04:46 +02:00 |
|
|
Pepijn
|
53577f5f1a
|
rename action_dim, state_dim to max_action_dim, max_state_dim
|
2025-09-17 16:34:07 +02:00 |
|
|
Pepijn
|
8c0cdb00a6
|
replace decorator test_pi0_openpi
|
2025-09-17 16:04:20 +02:00 |
|
|
Pepijn
|
c3b2fadb35
|
remove require_nightly_gpu
|
2025-09-17 15:56:39 +02:00 |
|
|
Pepijn
|
256b0e1e3c
|
feat: only run pi test on GPU
|
2025-09-17 15:55:58 +02:00 |
|
|
Pepijn
|
7aebc526b2
|
fix circular input
|
2025-09-17 15:30:47 +02:00 |
|
|
Pepijn
|
df78ea5390
|
add back init
|
2025-09-17 10:41:56 +02:00 |
|
|
Pepijn
|
c73b941809
|
add new models to modelcard
|
2025-09-17 09:37:34 +02:00 |
|
|
Pepijn
|
ac5f77ccbb
|
fix tests
|
2025-09-16 19:03:01 +02:00 |
|
|
Pepijn
|
3a6a2dc637
|
skip if no transformers
|
2025-09-16 18:34:52 +02:00 |
|
|
Pepijn
|
8f624f1c1e
|
cast float64 to float32 for mps
|
2025-09-16 16:48:08 +02:00 |
|
|
Pepijn
|
5924d4d9eb
|
remove todo
|
2025-09-16 15:15:24 +02:00 |
|
|
Pepijn
|
aaae109447
|
minor docs fixes and dependencies
|
2025-09-16 15:11:14 +02:00 |
|
|
Pepijn
|
f9af76ea50
|
change docs: finetune base model options
|
2025-09-16 14:42:31 +02:00 |
|
|
Pepijn
|
9ac16f2cd6
|
adapt docs pi05
|
2025-09-16 14:40:52 +02:00 |
|
|
Pepijn
|
3e097f5887
|
update docs
|
2025-09-16 14:36:12 +02:00 |
|
|
Pepijn
|
aeacb8dfa0
|
go back to python 3.10
|
2025-09-16 14:29:30 +02:00 |
|
|
Pepijn
|
0e0d6fbfc2
|
Add test to instatiate all base models
|
2025-09-16 13:31:29 +02:00 |
|
|
Pepijn
|
6aaeb7c13f
|
put tests in test folder
|
2025-09-16 13:27:04 +02:00 |
|
|
Pepijn
|
4e64503cf8
|
Merge branch 'main' into feat/add_pi
|
2025-09-16 11:32:26 +02:00 |
|
|
Pepijn
|
70624da239
|
Add docs
|
2025-09-16 10:09:42 +02:00 |
|
|
Michel Aractingi
|
847e74f628
|
Update dataset card by default (#1936)
* remove condition on model card update
|
2025-09-15 18:52:30 +02:00 |
|
|
Francesco Capuano
|
33cad37054
|
Add Streaming Dataset (#1613)
Co-authored-by: Michel Aractingi <michel.aractingi@huggingface.co>
|
2025-09-15 14:08:01 +02:00 |
|
|
Michel Aractingi
|
f55c6e89f0
|
Dataset v3 (#1412)
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
Co-authored-by: Remi Cadene <re.cadene@gmail.com>
Co-authored-by: Tavish <tavish9.chen@gmail.com>
Co-authored-by: fracapuano <francesco.capuano@huggingface.co>
Co-authored-by: CarolinePascal <caroline8.pascal@gmail.com>
|
2025-09-15 09:53:30 +02:00 |
|
|
Pepijn
|
d883c78a94
|
remove additional image augmentations, lerobot dataset already does this
|
2025-09-13 21:20:09 +02:00 |
|
|
Pepijn
|
d0d714be47
|
rename to loss
|
2025-09-13 16:15:29 +02:00 |
|
|
Pepijn
|
7d9b469eee
|
fix override self.pretrained_path = None overwrite
|
2025-09-13 14:50:43 +02:00 |
|
|
Pepijn
|
6db39cad58
|
temp: hardcode base model
|
2025-09-13 14:43:09 +02:00 |
|
|
Pepijn
|
af0676f99e
|
load from pretrained_path
|
2025-09-13 14:27:07 +02:00 |
|
|
Pepijn
|
b9df1a4ac5
|
use same name for action and state dim as lerobot pi0 and remove fixed image keys
|
2025-09-13 13:08:41 +02:00 |
|
|
Pepijn
|
5361346bec
|
Do not add model prefix to normalization
|
2025-09-13 11:25:26 +02:00 |
|
|
Pepijn
|
c5a029a28a
|
also compile forward method
|
2025-09-13 11:12:54 +02:00 |
|
|
Pepijn
|
c8163662ad
|
add preprocess tests
|
2025-09-12 21:41:25 +02:00 |
|
|
Pepijn
|
376cc772ff
|
fix from pretrained
|
2025-09-12 21:12:48 +02:00 |
|
|
Pepijn
|
d1eefd4e97
|
fix: remove unused param
|
2025-09-12 20:25:55 +02:00 |
|
|
Pepijn
|
7a03223693
|
use safeauto_docstring
|
2025-09-12 20:19:16 +02:00 |
|
|
Pepijn
|
f840d2e006
|
fix(modeling pi0): nit warning message
|
2025-09-12 20:06:06 +02:00 |
|
|
Pepijn
|
e94844fa59
|
revert to openpi transformer replace python 3.11
|
2025-09-12 20:00:21 +02:00 |
|
|
Pepijn
|
990f8e9cc9
|
update to python 3.11
|
2025-09-12 19:04:42 +02:00 |
|
|
Pepijn
|
6ce2a00135
|
also for pi05
|
2025-09-12 19:02:13 +02:00 |
|
|
Pepijn
|
bf90efa7e1
|
fix key match from pytorch state dict (similar keys to openpi implementation now)
|
2025-09-12 18:44:12 +02:00 |
|
|
Pepijn
|
dbe3406a69
|
add openpi image transforms for training and add more flexibility to _preprocess_images similar to lerobot pi0
|
2025-09-12 11:12:47 +02:00 |
|
|
Pepijn
|
1785767e61
|
clean up padding of state and action (more in line with lerobot pi0)
|
2025-09-12 10:38:24 +02:00 |
|
|
Pepijn
|
2234b851c0
|
rename action_horizon to chunk_size
|
2025-09-11 19:42:25 +02:00 |
|
|
Pepijn
|
b044f3104b
|
remove check
|
2025-09-11 11:03:41 +02:00 |
|
|
Pepijn
|
384ec52ec7
|
add pi05 to factory
|
2025-09-11 11:01:31 +02:00 |
|
|
Pepijn
|
8d1434c069
|
remove warning in config
|
2025-09-11 10:37:56 +02:00 |
|
|
Pepijn
|
f613a37cd2
|
add some comments, license and readme
|
2025-09-11 10:36:38 +02:00 |
|
|
Pepijn
|
494aa576b2
|
fix push to hub test
|
2025-09-11 09:18:20 +02:00 |
|