Spaces:
Runtime error
Runtime error
add push_to_hub example
Browse files
README.md
CHANGED
|
@@ -62,7 +62,7 @@ accelerate launch --config_file accelerate_local.yaml \
|
|
| 62 |
--mixed_precision no
|
| 63 |
```
|
| 64 |
|
| 65 |
-
#### Run training on local machine with `batch_size` of
|
| 66 |
|
| 67 |
```bash
|
| 68 |
accelerate launch --config_file accelerate_local.yaml \
|
|
@@ -76,7 +76,10 @@ accelerate launch --config_file accelerate_local.yaml \
|
|
| 76 |
--gradient_accumulation_steps 8 \
|
| 77 |
--learning_rate 1e-4 \
|
| 78 |
--lr_warmup_steps 500 \
|
| 79 |
-
--mixed_precision no
|
|
|
|
|
|
|
|
|
|
| 80 |
```
|
| 81 |
|
| 82 |
#### Run training on SageMaker.
|
|
|
|
| 62 |
--mixed_precision no
|
| 63 |
```
|
| 64 |
|
| 65 |
+
#### Run training on local machine with `batch_size` of 2 and `gradient_accumulation_steps` 8 to compensate, so that 256x256 resolution model fits on commercial grade GPU and push to hub.
|
| 66 |
|
| 67 |
```bash
|
| 68 |
accelerate launch --config_file accelerate_local.yaml \
|
|
|
|
| 76 |
--gradient_accumulation_steps 8 \
|
| 77 |
--learning_rate 1e-4 \
|
| 78 |
--lr_warmup_steps 500 \
|
| 79 |
+
--mixed_precision no \
|
| 80 |
+
--push_to_hub True \
|
| 81 |
+
--hub_model_id teticio/audio-diffusion-256 \
|
| 82 |
+
--hub_token $(cat $HOME/.huggingface/token)
|
| 83 |
```
|
| 84 |
|
| 85 |
#### Run training on SageMaker.
|