test

app.py

@@ -1,7 +1,7 @@
 import os
 if os.getenv('SPACES_ZERO_GPU') == "true":
     os.environ['SPACES_ZERO_GPU'] = "1"
-os.environ['K_DIFFUSION_USE_COMPILE'] = "1"
+os.environ['K_DIFFUSION_USE_COMPILE'] = "0"
 import spaces
 import cv2
 import gradio as gr
@@ -30,7 +30,7 @@ half = True if torch.cuda.is_available() else False
 upsampler = RealESRGANer(scale=4, model_path=realesr_model_path, model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
 
 pmrf = MMSERectifiedFlow.from_pretrained('ohayonguy/PMRF_blind_face_image_restoration').to(device)
-pmrf = torch.compile(pmrf, mode="reduce-overhead")
+
 face_helper_dummy = FaceRestoreHelper(
     1,
     face_size=512,

requirements.txt

@@ -1,6 +1,6 @@
--f https://download.pytorch.org/whl/cu124
+-f https://download.pytorch.org/whl/cu121
 torch==2.4.0
--f https://download.pytorch.org/whl/cu124
+-f https://download.pytorch.org/whl/cu121
 torchvision==0.19.0
 realesrgan==0.2.5
 numpy
@@ -22,4 +22,4 @@ timm
 torchmetrics
 torch-fidelity==0.3.0
 -f https://shi-labs.com/natten/wheels/
-natten==0.17.1+torch240cu124
+natten==0.17.1+torch240cu121