Skip to content

GitLab

"git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "68380ab8cd5a40a2b4e812cd9a649fe3838b682f"
Commit 494cfe54 authored by comfyanonymous's avatar comfyanonymous
Browse files

Prevent model_management from being loaded twice.

parent 07598e27
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
comfy_extras/nodes_upscale_model.py
View file @ 494cfe54
import os import os
from comfy_extras.chainner_models import model_loading from comfy_extras.chainner_models import model_loading
from comfy.sd import load_torch_file from comfy.sd import load_torch_file
import comfy.model_management import model_management
from nodes import filter_files_extensions, recursive_search, supported_ckpt_extensions from nodes import filter_files_extensions, recursive_search, supported_ckpt_extensions
import torch import torch
import comfy.utils import comfy.utils
...@@ -38,7 +38,7 @@ class ImageUpscaleWithModel: ...@@ -38,7 +38,7 @@ class ImageUpscaleWithModel:
CATEGORY = "image/upscaling" CATEGORY = "image/upscaling"
def upscale(self, upscale_model, image): def upscale(self, upscale_model, image):
device = comfy.model_management.get_torch_device() device = model_management.get_torch_device()
upscale_model.to(device) upscale_model.to(device)
in_img = image.movedim(-1,-3).to(device) in_img = image.movedim(-1,-3).to(device)
s = comfy.utils.tiled_scale(in_img, lambda a: upscale_model(a), tile_x=128 + 64, tile_y=128 + 64, overlap = 8, upscale_amount=upscale_model.scale) s = comfy.utils.tiled_scale(in_img, lambda a: upscale_model(a), tile_x=128 + 64, tile_y=128 + 64, overlap = 8, upscale_amount=upscale_model.scale)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment