Instructions to use lerobot/diffusion_pusht with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use lerobot/diffusion_pusht with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("lerobot/diffusion_pusht", dtype="auto") - Notebooks
- Google Colab
- Kaggle
Add use_amp for backwards compatibility
Browse files- config.yaml +1 -0
config.yaml
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
device: cuda
|
|
|
|
| 2 |
seed: 100000
|
| 3 |
dataset_repo_id: lerobot/pusht
|
| 4 |
training:
|
|
|
|
| 1 |
device: cuda
|
| 2 |
+
use_amp: false
|
| 3 |
seed: 100000
|
| 4 |
dataset_repo_id: lerobot/pusht
|
| 5 |
training:
|