minor

sayakpaul · sayakpaul · commit 4509f3492021 · 2024-12-13T14:24:40.000+05:30
diff --git a/tests/lora/test_lora_layers_flux.py b/tests/lora/test_lora_layers_flux.py
@@ -446,7 +446,7 @@ def test_lora_unload_with_parameter_expanded_shapes(self):
             f"Expected {num_channels_without_control} channels in the modified transformer but has {transformer.config.in_channels=}",
         )
 
-        # This should be initialize with a Flux pipeline variant that doesn't accept `control_image`.
+        # This should be initialized with a Flux pipeline variant that doesn't accept `control_image`.
         components["transformer"] = transformer
         pipe = FluxPipeline(**components)
         pipe = pipe.to(torch_device)

Original file line number	Diff line number	Diff line change
`@@ -446,7 +446,7 @@ def test_lora_unload_with_parameter_expanded_shapes(self):`
`446`	`446`	`f"Expected {num_channels_without_control} channels in the modified transformer but has {transformer.config.in_channels=}",`
`447`	`447`	`)`
`448`	`448`
`449`		- # This should be initialize with a Flux pipeline variant that doesn't accept `control_image`.
	`449`	+ # This should be initialized with a Flux pipeline variant that doesn't accept `control_image`.
`450`	`450`	`components["transformer"] = transformer`
`451`	`451`	`pipe = FluxPipeline(**components)`
`452`	`452`	`pipe = pipe.to(torch_device)`