File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change 3737
3838class DreamBoothLoRAHiDreamImage (ExamplesTestsAccelerate ):
3939 instance_data_dir = "docs/source/en/imgs"
40+ instance_prompt = "photo"
4041 pretrained_model_name_or_path = "hf-internal-testing/tiny-hidream-i1-pipe"
4142 text_encoder_4_path = "hf-internal-testing/tiny-random-LlamaForCausalLM"
4243 tokenizer_4_path = "hf-internal-testing/tiny-random-LlamaForCausalLM"
@@ -187,6 +188,7 @@ def test_dreambooth_lora_with_metadata(self):
187188 { self .script_path }
188189 --pretrained_model_name_or_path { self .pretrained_model_name_or_path }
189190 --instance_data_dir { self .instance_data_dir }
191+ --instance_prompt { self .instance_prompt }
190192 --resolution 64
191193 --train_batch_size 1
192194 --gradient_accumulation_steps 1
@@ -227,6 +229,7 @@ def test_dreambooth_lora_hidream_checkpointing_checkpoints_total_limit_removes_m
227229 --pretrained_text_encoder_4_name_or_path { self .text_encoder_4_path }
228230 --pretrained_tokenizer_4_name_or_path { self .tokenizer_4_path }
229231 --instance_data_dir={ self .instance_data_dir }
232+ --instance_prompt { self .instance_prompt }
230233 --output_dir={ tmpdir }
231234 --resolution=32
232235 --train_batch_size=1
@@ -247,6 +250,7 @@ def test_dreambooth_lora_hidream_checkpointing_checkpoints_total_limit_removes_m
247250 --pretrained_text_encoder_4_name_or_path { self .text_encoder_4_path }
248251 --pretrained_tokenizer_4_name_or_path { self .tokenizer_4_path }
249252 --instance_data_dir={ self .instance_data_dir }
253+ --instance_prompt { self .instance_prompt }
250254 --output_dir={ tmpdir }
251255 --resolution=32
252256 --train_batch_size=1
You can’t perform that action at this time.
0 commit comments