inoid commited on
Commit
c704363
·
1 Parent(s): 087e3b8

Fix error in generate

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ def train_model(*inputs):
47
  train_text_encoder = True,
48
  #stop_text_encoder_training = stptxt,
49
  save_n_steps = 0,
50
- pretrained_model_name_or_path = model_to_load,
51
  instance_data_dir="instance_images",
52
  #class_data_dir=class_data_dir,
53
  output_dir="output_model",
 
47
  train_text_encoder = True,
48
  #stop_text_encoder_training = stptxt,
49
  save_n_steps = 0,
50
+ #pretrained_model_name_or_path = model_to_load,
51
  instance_data_dir="instance_images",
52
  #class_data_dir=class_data_dir,
53
  output_dir="output_model",