-
Notifications
You must be signed in to change notification settings - Fork 3
Description
The Pruna install ran without error.
starting comfy as instructed: python3 main.py --disable-cuda-malloc --gpu-only
Not sure what I'm missing, please help!
############################################
PyTorch version 2.5.1+cu124 available.
ComfyUI_pruna: Pruna node import failed.
Traceback (most recent call last):
File "/mnt/d/ComfyUI/custom_nodes/ComfyUI_pruna/init.py", line 10, in
from .node import CompileModel
File "/mnt/d/ComfyUI/custom_nodes/ComfyUI_pruna/node.py", line 4, in
from pruna import SmashConfig, smash
File "pruna/init.py", line 3, in init pruna
File "pruna/smash.py", line 6, in init pruna.smash
File "pruna/algorithms/compilation/compile.py", line 19, in init pruna.algorithms.compilation.compile
File "pruna/algorithms/compilation/torch_compile.py", line 29, in pruna.algorithms.compilation.torch_compile.TorchCompileCompiler.init
File "pruna/algorithms/compilation/PrunaCompiler.py", line 13, in pruna.algorithms.compilation.PrunaCompiler.PrunaCompiler.init
File "pruna/algorithms/PrunaBase.py", line 11, in pruna.algorithms.PrunaBase.PrunaSmasherBase.init
File "pruna/algorithms/compilation/torch_compile.py", line 52, in pruna.algorithms.compilation.torch_compile.TorchCompileCompiler.get_hyperparameters
File "ConfigSpace/hyperparameters/categorical.pyx", line 81, in ConfigSpace.hyperparameters.categorical.CategoricalHyperparameter.init
TypeError: Choice 'None' is not supported
Traceback (most recent call last):
File "/mnt/d/ComfyUI/nodes.py", line 2001, in load_custom_node
module_spec.loader.exec_module(module)
File "", line 883, in exec_module
File "", line 241, in _call_with_frames_removed
File "/mnt/d/ComfyUI/custom_nodes/ComfyUI_pruna/init.py", line 26, in
raise Exception("import failed")
Exception: import failed