AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

reduce lora rank.

Browse Source
This commit is contained in:
sayakpaul
2023-11-14 13:55:24 +05:30
parent 8c4eaf6715
commit cece781921
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/test_examples.py
View File

@@ -1681,13 +1681,14 @@ class ExamplesTestsAccelerate(unittest.TestCase):
)
self.assertTrue(starts_with_unet)
def test_text_to_image_lcm__lora_sdxl(self):
def test_text_to_image_lcm_lora_sdxl(self):
with tempfile.TemporaryDirectory() as tmpdir:
test_args = f"""
examples/consistency_distillation/train_lcm_distill_lora_sdxl.py
--pretrained_teacher_model hf-internal-testing/tiny-stable-diffusion-xl-pipe
--dataset_name hf-internal-testing/dummy_image_text_data
--resolution 64
--lora_rank 4
--train_batch_size 1
--gradient_accumulation_steps 1
--max_train_steps 2