Skip to content

Commit 9f4fcee

Browse files
billwuhaobillwuhao
committed
v1.0.0
1 parent 10fed8c commit 9f4fcee

11 files changed

+3167
-1
lines changed

Diff for: ‎.github/workflows/publish_action.yml

+22
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
name: Publish to Comfy registry
2+
on:
3+
workflow_dispatch:
4+
push:
5+
branches:
6+
- master
7+
- main
8+
paths:
9+
- "pyproject.toml"
10+
11+
jobs:
12+
publish-node:
13+
name: Publish Custom Node to registry
14+
runs-on: ubuntu-latest
15+
steps:
16+
- name: Check out code
17+
uses: actions/checkout@v4
18+
- name: Publish Custom Node
19+
uses: Comfy-Org/publish-node-action@main
20+
with:
21+
## Add your own personal access token to your Github Repository secrets and reference it here.
22+
personal_access_token: ${{ secrets.REGISTRY_ACCESS_TOKEN }}

Diff for: ‎NotaGenNode.py

+370
Large diffs are not rendered by default.

Diff for: ‎README-en.md

+13
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
[中文](README.md) | [English](README-en.md)
2+
3+
# Symbolic Music Generation, NotaGen node for ComfyUI.
4+
5+
![image](https://github.com/billwuhao/ComfyUI_KokoroTTS_MW/blob/master/images/2025-03-10_06-24-03.png)
6+
7+
Download the model to `ComfyUI\models\TTS\NotaGen` and rename it as required:
8+
9+
[NotaGen-X](https://huggingface.co/ElectricAlexis/NotaGen/blob/main/weights_notagenx_p_size_16_p_length_1024_p_layers_20_h_size_1280.pth) → `notagenx.pth`
10+
[NotaGen-small](https://huggingface.co/ElectricAlexis/NotaGen/blob/main/weights_notagen_pretrain_p_size_16_p_length_2048_p_layers_12_c_layers_3_h_size_768_lr_0.0002_batch_8.pth) → `notagen_small.pth`
11+
[NotaGen-medium](https://huggingface.co/ElectricAlexis/NotaGen/blob/main/weights_notagen_pretrain_p_size_16_p_length_2048_p_layers_16_c_layers_3_h_size_1024_lr_0.0001_batch_4.pth) → `notagen_medium.pth`
12+
[NotaGen-large](https://huggingface.co/ElectricAlexis/NotaGen/blob/main/weights_notagen_pretrain_p_size_16_p_length_1024_p_layers_20_c_layers_6_h_size_1280_lr_0.0001_batch_4.pth) → `notagen_large.pth`
13+

Diff for: ‎README.md

+13-1
Original file line numberDiff line numberDiff line change
@@ -1 +1,13 @@
1-
# ComfyUI_NotaGen
1+
[中文](README.md) | [English](README-en.md)
2+
3+
# 符号音乐生成. NotaGen 的 ComfyUI 节点.
4+
5+
![image](https://github.com/billwuhao/ComfyUI_KokoroTTS_MW/blob/master/images/2025-03-10_06-24-03.png)
6+
7+
将模型下载放到 `ComfyUI\models\TTS\NotaGen` 下, 并按要求重命名:
8+
9+
[NotaGen-X](https://huggingface.co/ElectricAlexis/NotaGen/blob/main/weights_notagenx_p_size_16_p_length_1024_p_layers_20_h_size_1280.pth) → `notagenx.pth`
10+
[NotaGen-small](https://huggingface.co/ElectricAlexis/NotaGen/blob/main/weights_notagen_pretrain_p_size_16_p_length_2048_p_layers_12_c_layers_3_h_size_768_lr_0.0002_batch_8.pth) → `notagen_small.pth`
11+
[NotaGen-medium](https://huggingface.co/ElectricAlexis/NotaGen/blob/main/weights_notagen_pretrain_p_size_16_p_length_2048_p_layers_16_c_layers_3_h_size_1024_lr_0.0001_batch_4.pth) → `notagen_medium.pth`
12+
[NotaGen-large](https://huggingface.co/ElectricAlexis/NotaGen/blob/main/weights_notagen_pretrain_p_size_16_p_length_1024_p_layers_20_c_layers_6_h_size_1280_lr_0.0001_batch_4.pth) → `notagen_large.pth`
13+

Diff for: ‎__init__.py

+3
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
from .NotaGenNode import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
2+
3+
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]

Diff for: ‎abc2xml.py

+2,252
Large diffs are not rendered by default.

Diff for: ‎config.py

+30
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
# Configurations for model
2+
nota_lx = {
3+
"PATCH_STREAM": True, # Stream training / inference
4+
"PATCH_SIZE": 16, # Patch Size
5+
"PATCH_LENGTH": 1024, # Patch Length
6+
"CHAR_NUM_LAYERS": 6, # Number of layers in the decoder
7+
"PATCH_NUM_LAYERS": 20, # Number of layers in the encoder
8+
"HIDDEN_SIZE": 1280, # Hidden Size
9+
"PATCH_SAMPLING_BATCH_SIZE": 0 # Batch size for patch during training, 0 for full conaudio
10+
}
11+
12+
nota_small = {
13+
"PATCH_STREAM": True, # Stream training / inference
14+
"PATCH_SIZE": 16, # Patch Size
15+
"PATCH_LENGTH": 2048, # Patch Length
16+
"CHAR_NUM_LAYERS": 3, # Number of layers in the decoder
17+
"PATCH_NUM_LAYERS": 12, # Number of layers in the encoder
18+
"HIDDEN_SIZE": 768, # Hidden Size
19+
"PATCH_SAMPLING_BATCH_SIZE": 0 # Batch size for patch during training, 0 for full conaudio
20+
}
21+
22+
nota_medium = {
23+
"PATCH_STREAM": True, # Stream training / inference
24+
"PATCH_SIZE": 16, # Patch Size
25+
"PATCH_LENGTH": 2048, # Patch Length
26+
"CHAR_NUM_LAYERS": 3, # Number of layers in the decoder
27+
"PATCH_NUM_LAYERS": 16, # Number of layers in the encoder
28+
"HIDDEN_SIZE": 1024, # Hidden Size
29+
"PATCH_SAMPLING_BATCH_SIZE": 0 # Batch size for patch during training, 0 for full conaudio
30+
}

Diff for: ‎images/2025-03-10_06-24-03.png

36.3 KB
Loading

Diff for: ‎pyproject.toml

+14
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
[project]
2+
name = "notagen-mw"
3+
description = "Symbolic Music Generation, NotaGen node for ComfyUI."
4+
version = "1.0.0"
5+
license = {file = "LICENSE"}
6+
7+
[project.urls]
8+
Repository = "https://github.com/billwuhao/ComfyUI_NotaGen"
9+
# Used by Comfy Registry https://comfyregistry.org
10+
11+
[tool.comfy]
12+
PublisherId = "mw"
13+
DisplayName = "ComfyUI_NotaGen"
14+
Icon = "https://github.com/billwuhao/aiart.website/blob/master/hb.png"

Diff for: ‎requirements.txt

+7
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
# transformers==4.40.0
2+
# numpy==1.26.4
3+
# gradio==5.17.1
4+
wandb>=0.17.2
5+
abctoolkit>=0.0.6
6+
samplings>=0.1.7
7+
pyparsing>=3.2.1

0 commit comments

Comments
 (0)