Spaces:
Runtime error
Runtime error
Fix error in generate
Browse files
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",
|