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.
|