Men1scus commited on
Commit
05c79ef
·
1 Parent(s): aad20a1

fix: Ensure input image is converted to RGB format in process_sr function

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -328,6 +328,9 @@ def process_sr(
328
  transforms.Resize(process_size, interpolation=transforms.InterpolationMode.BILINEAR),
329
  ])
330
 
 
 
 
331
  seed_everything(seed)
332
  generator = torch.Generator(device=dit4sr_device)
333
  generator.manual_seed(seed)
 
328
  transforms.Resize(process_size, interpolation=transforms.InterpolationMode.BILINEAR),
329
  ])
330
 
331
+ if input_image.mode != 'RGB':
332
+ input_image = input_image.convert('RGB')
333
+
334
  seed_everything(seed)
335
  generator = torch.Generator(device=dit4sr_device)
336
  generator.manual_seed(seed)