Skip to content

Commit 98cebb5

Browse files
committed
Fix some imports
1 parent 13a6d33 commit 98cebb5

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/inference_core_nodes/prompt_expansion/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
def _add_folder_paths():
1111
from pathlib import Path
12-
from inference.inference_server import folder_paths
12+
import folder_paths
1313

1414
folder_paths.add_model_folder_path(
1515
MODEL_FOLDER_NAME,

src/inference_core_nodes/prompt_expansion/prompt_expansion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@
77
import torch
88
from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed
99

10+
import folder_paths
1011
from comfy import model_management
11-
from comfy import folder_paths
1212
from comfy.model_patcher import ModelPatcher
1313
from .util import join_prompts, remove_empty_str
1414

0 commit comments

Comments
 (0)