Skip to content

Commit 3b9dad8

Browse files
committed
fix test failures.
1 parent 6c8314d commit 3b9dad8

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

examples/dreambooth/test_dreambooth_lora_hidream.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@
3737

3838
class 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

0 commit comments

Comments
 (0)