From e9b06d5d8bc902e10e165d3df9f6f7ae0bdf9dd5 Mon Sep 17 00:00:00 2001 From: HowToSD Date: Fri, 14 Mar 2025 19:44:02 -0700 Subject: [PATCH] Fixed doc generator to generate doc for a few nodes which did not have doc --- docs/reference/ptn_chained_model.md | 24 ++++++++++++++++++++++++ docs/reference/pto_simple.md | 20 ++++++++++++++++++++ 2 files changed, 44 insertions(+) create mode 100644 docs/reference/ptn_chained_model.md create mode 100644 docs/reference/pto_simple.md diff --git a/docs/reference/ptn_chained_model.md b/docs/reference/ptn_chained_model.md new file mode 100644 index 0000000..ca2ae03 --- /dev/null +++ b/docs/reference/ptn_chained_model.md @@ -0,0 +1,24 @@ +# Ptn Chained Model +Constructs a chained PyTorch model. + +### Fields: +- model_a: First model in the chain. +- model_b: Second model in the chain. +- closure: Optional differentiable function to be called at the end of forward. + +## Input +| Name | Data type | +|---|---| +| model_a | Ptmodel | +| model_b | Ptmodel | +| closure | Ptcallable | + +## Output +| Data type | +|---| +| Ptmodel | + +
+Category: PyTorch wrapper - Training + +ComfyUI Pt Wrapper Node Reference. © 2025 Hide Inada (HowToSD.com). All rights reserved. diff --git a/docs/reference/pto_simple.md b/docs/reference/pto_simple.md new file mode 100644 index 0000000..264f9f1 --- /dev/null +++ b/docs/reference/pto_simple.md @@ -0,0 +1,20 @@ +# Instantiates the most basic optimizer to update W using the below formula +$$ +W -= lr * G +$$ + +## Input +| Name | Data type | +|---|---| +| model | Ptmodel | +| learning_rate | Float | + +## Output +| Data type | +|---| +| Ptoptimizer | + +
+Category: PyTorch wrapper - Training + +ComfyUI Pt Wrapper Node Reference. © 2025 Hide Inada (HowToSD.com). All rights reserved.