diff --git a/REAL-Video-Enhancer.py b/REAL-Video-Enhancer.py index 705dcfcc..acada009 100644 --- a/REAL-Video-Enhancer.py +++ b/REAL-Video-Enhancer.py @@ -150,14 +150,11 @@ def __init__(self): for line in self.fullOutput.lower().split("\n"): if "half precision support:" in line: halfPrecisionSupport = "true" in line - if "gmfss support:" in line: - gmfssSupport = "true" in line settings = Settings() settings.readSettings() self.settings = settings self.processTab = ProcessTab( parent=self, - gmfssSupport=gmfssSupport, ) self.homeTab = HomeTab(parent=self) self.downloadTab = DownloadTab(parent=self) diff --git a/backend/rve-backend.py b/backend/rve-backend.py index f6628b9e..001f089f 100644 --- a/backend/rve-backend.py +++ b/backend/rve-backend.py @@ -48,8 +48,6 @@ def __init__(self): ) else: half_prec_supp = False - gmfss_supp = False - vram = 0 availableBackends = [] printMSG = "" @@ -75,18 +73,13 @@ def __init__(self): availableBackends.append("pytorch (cuda)") printMSG += f"PyTorch Version: {torch.__version__}\n" half_prec_supp = check_bfloat16_support() - gmfss_supp = checkForGMFSS() - vram = get_pytorch_vram() if checkForPytorchROCM(): availableBackends.append("pytorch (rocm)") import torch printMSG += f"PyTorch Version: {torch.__version__}\n" half_prec_supp = check_bfloat16_support() - try: - vram = get_pytorch_vram() - except Exception: - vram = 0 + if checkForNCNN(): availableBackends.append("ncnn") printMSG += f"NCNN Version: 20220729\n" @@ -98,11 +91,7 @@ def __init__(self): printMSG += f"ONNXruntime Version: {ort.__version__}\n" half_prec_supp = checkForDirectMLHalfPrecisionSupport() printMSG += f"Half precision support: {half_prec_supp}\n" - printMSG += f"GMFSS support: {gmfss_supp}\n" - if not gmfss_supp and "pytorch (cuda)" in availableBackends: - printMSG += "Please install CUDA to enable GMFSS\n" - if vram != 0: - printMSG += f"VRAM: {vram}mb\n" + print("Available Backends: " + str(availableBackends)) print(printMSG) diff --git a/src/ui/ProcessTab.py b/src/ui/ProcessTab.py index 9ac54b03..3e1343f9 100644 --- a/src/ui/ProcessTab.py +++ b/src/ui/ProcessTab.py @@ -40,7 +40,7 @@ class ProcessTab: - def __init__(self, parent, gmfssSupport: bool): + def __init__(self, parent): self.parent = parent self.imagePreviewSharedMemoryID = "/image_preview" + str(os.getpid()) self.renderTextOutputList = None @@ -48,7 +48,6 @@ def __init__(self, parent, gmfssSupport: bool): self.animationHandler = AnimationHandler() self.tileUpAnimationHandler = AnimationHandler() self.tileDownAnimationHandler = AnimationHandler() - self.gmfssSupport = gmfssSupport # encoder dict # key is the name in RVE gui # value is the encoder used @@ -97,15 +96,7 @@ def populateModels(self, backend) -> dict: ["None"] + list(interpolateModels.keys()) ) self.parent.upscaleModelComboBox.addItems(["None"] + list(upscaleModels.keys())) - if not self.gmfssSupport: - # Disable specific options based on the selected text - for i in range(self.parent.interpolateModelComboBox.count()): - if "GMFSS" in self.parent.interpolateModelComboBox.itemText( - i - ): # hacky solution, just straight copy pasted - self.parent.interpolateModelComboBox.model().item(i).setEnabled( - self.gmfssSupport - ) + def onTilingSwitch(self): if self.parent.tilingCheckBox.isChecked():