mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user