lvkaokao commited on
Commit
9be9b07
1 Parent(s): bfb5b0d

update app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -112,6 +112,7 @@ def pad_image(image):
112
  return new_image
113
 
114
  def train(*inputs):
 
115
  if is_shared_ui:
116
  raise gr.Error("This Space only works in duplicated instances")
117
 
@@ -184,7 +185,6 @@ def train(*inputs):
184
  instance_prompt="",
185
  seed=42,
186
  resolution=512,
187
- mixed_precision="fp16",
188
  train_batch_size=1,
189
  gradient_accumulation_steps=1,
190
  use_8bit_adam=True,
@@ -567,4 +567,4 @@ with gr.Blocks(css=css) as demo:
567
  #Checks if the training is running
568
  demo.load(fn=check_status, inputs=top_description, outputs=[top_description, try_your_model, push_to_hub, result, convert_button], queue=False, show_progress=False)
569
 
570
- demo.queue(default_enabled=False).launch(debug=True)
 
112
  return new_image
113
 
114
  def train(*inputs):
115
+ print(inputs)
116
  if is_shared_ui:
117
  raise gr.Error("This Space only works in duplicated instances")
118
 
 
185
  instance_prompt="",
186
  seed=42,
187
  resolution=512,
 
188
  train_batch_size=1,
189
  gradient_accumulation_steps=1,
190
  use_8bit_adam=True,
 
567
  #Checks if the training is running
568
  demo.load(fn=check_status, inputs=top_description, outputs=[top_description, try_your_model, push_to_hub, result, convert_button], queue=False, show_progress=False)
569
 
570
+ demo.queue(default_enabled=False).launch(debug=True)