Update inference_codeformer_cpu.py

inference_codeformer_cpu.py

@@ -23,23 +23,11 @@ pretrain_model_url = {
 }
 
 def set_realesrgan(args):
-    """
-    CPU esetén letiltjuk a RealESRGAN-t (visszatér None-nal).
-    Ha valaki mégis GPU-val futtatná és CUDA elérhető, itt létrehozzuk az upsampler-t.
-    (A környezetünk CPU-only, így ez a függvény tipikusan None-t ad vissza.)
-    """
-    # importok helyben, mert ez a funkció opcionális
     from basicsr.archs.rrdbnet_arch import RRDBNet
     from basicsr.utils.realesrgan_utils import RealESRGANer
 
-    # CPU környezetben nem használunk fp16-et
-    use_half = False
-
-    if not torch.cuda.is_available():
-        warnings.warn('RealESRGAN: CUDA nem elérhető; a háttér feljavítás CPU-n letiltva.', RuntimeWarning)
-        return None
+    device = torch.device("cpu")
 
-    # Ha itt valaki GPU-n futtatja (nem jellemző ebben a CPU-only scriptben):
     model = RRDBNet(
         num_in_ch=3,
         num_out_ch=3,
@@ -48,17 +36,21 @@ def set_realesrgan(args):
         num_grow_ch=32,
         scale=2,
     )
+
     upsampler = RealESRGANer(
         scale=2,
         model_path="https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/RealESRGAN_x2plus.pth",
         model=model,
-        tile=args.bg_tile,
-        tile_pad=40,
+        tile=min(args.bg_tile, 200),   # COLAB CPU LIMIT
+        tile_pad=20,
         pre_pad=0,
-        half=use_half
+        half=False,
+        device=device
     )
+
     return upsampler
 
+
 if __name__ == '__main__':
     # explicit CPU device
     device = torch.device("cpu")