fix for 73c3e11

2023-04-30 09:17:50 +09:00
parent 67c78d7d1c
commit c991773978
1 changed files with 1 additions and 1 deletions
@@ -2,7 +2,7 @@ import re
 from itertools import product
 from typing import Callable, List, Dict, Any, Union, Tuple, cast
 import torch
-import model_management # type: ignore
+from comfy import model_management
 import comfy.samplers
 from nodes import common_ksampler
 from comfy.sd import ModelPatcher