hysts HF staff commited on
Commit
87968ab
1 Parent(s): db0cd98
Files changed (1) hide show
  1. app_training.py +25 -12
app_training.py CHANGED
@@ -113,18 +113,31 @@ def create_training_demo(trainer: Trainer,
113
 
114
  if pipe is not None:
115
  run_button.click(fn=pipe.clear)
116
- run_button.click(
117
- fn=trainer.run,
118
- inputs=[
119
- training_video, training_prompt, output_model_name,
120
- delete_existing_repo, validation_prompt, base_model,
121
- resolution, num_training_steps, learning_rate,
122
- gradient_accumulation, seed, fp16, use_8bit_adam,
123
- checkpointing_steps, validation_epochs, upload_to_hub,
124
- use_private_repo, delete_existing_repo, upload_to,
125
- remove_gpu_after_training, input_token
126
- ],
127
- outputs=output_message)
 
 
 
 
 
 
 
 
 
 
 
 
 
128
  return demo
129
 
130
 
 
113
 
114
  if pipe is not None:
115
  run_button.click(fn=pipe.clear)
116
+ run_button.click(fn=trainer.run,
117
+ inputs=[
118
+ training_video,
119
+ training_prompt,
120
+ output_model_name,
121
+ delete_existing_repo,
122
+ validation_prompt,
123
+ base_model,
124
+ resolution,
125
+ num_training_steps,
126
+ learning_rate,
127
+ gradient_accumulation,
128
+ seed,
129
+ fp16,
130
+ use_8bit_adam,
131
+ checkpointing_steps,
132
+ validation_epochs,
133
+ upload_to_hub,
134
+ use_private_repo,
135
+ delete_existing_repo,
136
+ upload_to,
137
+ remove_gpu_after_training,
138
+ input_token,
139
+ ],
140
+ outputs=output_message)
141
  return demo
142
 
143