nodes/kijai-ComfyUI-LuminaWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rename

Browse Source
This commit is contained in:
kijai
2024-06-17 11:56:03 +03:00
parent a18eebf516
commit 8fa27595a5
4 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/__init__.py → lumina_models/__init__.py
View File
models/components.py → lumina_models/components.py
View File
models/nextdit.py → lumina_models/nextdit.py
View File
nodes.py
+2 -2
View File
@@ -10,7 +10,7 @@ import folder_paths
script_directory = os.path.dirname(os.path.abspath(__file__)) script_directory = os.path.dirname(os.path.abspath(__file__))
sys.path.append(script_directory) sys.path.append(script_directory)
import models import lumina_models
from transport import ODE from transport import ODE
from transformers import AutoModel, AutoTokenizer from transformers import AutoModel, AutoTokenizer
@@ -55,7 +55,7 @@ class DownloadAndLoadLuminaModel:
train_args = torch.load(os.path.join(model_path, "model_args.pth")) train_args = torch.load(os.path.join(model_path, "model_args.pth"))
model = models.__dict__[train_args.model](qk_norm=train_args.qk_norm, cap_feat_dim=2048) model = lumina_models.__dict__[train_args.model](qk_norm=train_args.qk_norm, cap_feat_dim=2048)
model.eval().to(dtype) model.eval().to(dtype)
sd = load_torch_file(os.path.join(model_path, "consolidated.00-of-01.safetensors")) sd = load_torch_file(os.path.join(model_path, "consolidated.00-of-01.safetensors"))