update

requirements.txt

@@ -13,6 +13,6 @@ config==0.5.1
 einops==0.7.0
 onnxruntime==1.16.2
 huggingface_hub==0.25.0 
-torchvision==0.14.0
-torch
+torchvision==0.22.1
+torch==2.7.1
 basicsr

run/gradio_ootd.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import os
-os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:1024"
+os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:1024,expandable_segments:True"
 from pathlib import Path
 import sys
 import torch