teticio commited on
Commit
d73aa64
1 Parent(s): 633cada

add push_to_hub example

Browse files
Files changed (1) hide show
  1. README.md +5 -2
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 1 and `gradient_accumulation_steps` 16 to compensate, so that 256x256 resolution model fits on commercial grade GPU.
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.