@@ -134,7 +134,7 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step,
134134
135135    for  validation_prompt , validation_image  in  zip (validation_prompts , validation_images ):
136136        validation_image  =  Image .open (validation_image ).convert ("RGB" )
137-          
137+ 
138138        try :
139139            interpolation  =  getattr (transforms .InterpolationMode , args .image_interpolation_mode .upper ())
140140        except  (AttributeError , KeyError ):
@@ -145,11 +145,13 @@ def log_validation(vae, unet, controlnet, args, accelerator, weight_dtype, step,
145145                f"Interpolation mode { args .image_interpolation_mode }  
146146                f"Please select one of the following: { ', ' .join (supported_interpolation_modes )}  
147147            )
148-             
149-         transform  =  transforms .Compose ([
150-             transforms .Resize (args .resolution , interpolation = interpolation ),
151-             transforms .CenterCrop (args .resolution ),
152-         ])
148+ 
149+         transform  =  transforms .Compose (
150+             [
151+                 transforms .Resize (args .resolution , interpolation = interpolation ),
152+                 transforms .CenterCrop (args .resolution ),
153+             ]
154+         )
153155        validation_image  =  transform (validation_image )
154156
155157        images  =  []
@@ -767,7 +769,7 @@ def prepare_train_dataset(dataset, accelerator):
767769            f"Interpolation mode { args .image_interpolation_mode }  
768770            f"Please select one of the following: { ', ' .join (supported_interpolation_modes )}  
769771        )
770-      
772+ 
771773    image_transforms  =  transforms .Compose (
772774        [
773775            transforms .Resize (args .resolution , interpolation = interpolation_mode ),
0 commit comments