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)
|