ComfyUI Node: NNT Save Model

Authored by inventorado

Created

Updated

67 stars

Category

NNT Neural Network Toolkit/Models

Inputs

MODEL MODEL
filename STRING
model_path STRING
save_format
  • PyTorch Model
  • State Dict
  • TorchScript
  • ONNX
  • TorchScript Mobile
  • Quantized
  • SafeTensors
save_optimizer
  • True
  • False
optimizer
  • Adadelta
  • Adagrad
  • Adam
  • AdamW
  • SparseAdam
  • Adamax
  • ASGD
  • NAdam
  • RAdam
  • RMSprop
  • Rprop
  • SGD
  • None
quantization_type
  • dynamic
  • static
  • none
quantization_bits INT

Outputs

MODEL

STRING

Extension: ComfyUI Neural Network Toolkit NNT

Neural Network Toolkit (NNT) for ComfyUI is an extensive set of custom ComfyUI nodes for designing, training, and fine-tuning neural networks. This toolkit allows defining models, layers, training workflows, transformers, and tensor operations in a visual manner using nodes.

Authored by inventorado

View Nodes

Run ComfyUI workflows in the Cloud!

No downloads or installs are required. Pay only for active GPU usage, not idle time. No complex setups and dependency issues

Learn more