seedvr requirements

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-10-13 10:03:59 -04:00
parent d4d24214b3
commit 32014fbb9d
+2
View File
@@ -4,6 +4,7 @@ import numpy as np
import torch
from PIL import Image
from torchvision.transforms import ToPILImage
from installer import install
from modules import devices
from modules.shared import opts, log
from modules.upscaler import Upscaler, UpscalerData
@@ -32,6 +33,7 @@ class UpscalerSeedVR(Upscaler):
def load_model(self, path: str):
model_name = MODELS_MAP.get(path, None)
if (self.model is None) or (self.model_loaded != model_name):
install('rotary_embedding_torch')
log.debug(f'Upscaler loading: name="{self.name}" model="{model_name}"')
t0 = time.time()
from modules.seedvr.src.core.model_manager import configure_runner