From 019a2d3525d161d969c379257387a2210352450e Mon Sep 17 00:00:00 2001 From: kijai Date: Thu, 23 Nov 2023 16:21:50 +0200 Subject: [PATCH] Fixes for portable install --- comfyui_svd.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/comfyui_svd.py b/comfyui_svd.py index 61114c5..a1ba649 100644 --- a/comfyui_svd.py +++ b/comfyui_svd.py @@ -4,6 +4,7 @@ import math import torch import importlib import comfy.model_management +import os def get_obj_from_str(string, reload=False, invalidate_cache=True): module, cls = string.rsplit(".", 1) @@ -126,8 +127,9 @@ class SVDimg2vid: torch.cuda.ipc_collect() device: str = "cuda" - - model_config = f"custom_nodes/ComfyUI-SVD/svd/configs/{version}.yaml" + + script_directory = os.path.dirname(os.path.abspath(__file__)) + model_config = f"{script_directory}/svd/configs/{version}.yaml" model = load_model( model_config,