Spaces:

realfill-library
/

RealFill-Training-UI

Runtime error

thuanz123 commited on Oct 22, 2023

Commit

8dbbb39

1 Parent(s): ffa45e8

Update train_realfill.py

Files changed (1) hide show

train_realfill.py CHANGED Viewed

@@ -124,7 +124,7 @@ def log_validation(
     generator = None if args.seed is None else torch.Generator(device=accelerator.device).manual_seed(args.seed)
     target_dir = Path(args.train_data_dir) / "target"
-    target_image, target_mask = target_dir / "target.png", target_dir / "mask.png"
     image, mask_image = Image.open(target_image), Image.open(target_mask)
     if image.mode != "RGB":

     generator = None if args.seed is None else torch.Generator(device=accelerator.device).manual_seed(args.seed)
     target_dir = Path(args.train_data_dir) / "target"
+    target_image, target_mask = target_dir / "target.jpg", target_dir / "mask.jpg"
     image, mask_image = Image.open(target_image), Image.open(target_mask)
     if image.mode != "RGB":