Spaces:
Running
on
Zero
Running
on
Zero
Men1scus
commited on
Commit
·
05c79ef
1
Parent(s):
aad20a1
fix: Ensure input image is converted to RGB format in process_sr function
Browse files
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)
|