xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-04 10:17:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

linting

Browse Source
This commit is contained in:
Kohaku-Blueleaf 2025-04-22 15:14:26 +08:00
parent dc74839d2d
commit c8bd95a393
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/weight_adapter/__init__.py
View File

@ -11,3 +11,9 @@ adapters: list[type[WeightAdapterBase]] = [
LoKrAdapter, LoKrAdapter,
GLoRAAdapter, GLoRAAdapter,
] ]
__all__ = [
"WeightAdapterBase",
"WeightAdapterTrainBase",
"adapters"
] + [a.__name__ for a in adapters]

5
comfy_extras/nodes_train.py
View File

@ -1,7 +1,6 @@
import datetime import datetime
import json import json
import logging import logging
import math
import os import os
import numpy as np import numpy as np
@ -17,7 +16,7 @@ import folder_paths
import node_helpers import node_helpers
from comfy.cli_args import args from comfy.cli_args import args
from comfy.comfy_types.node_typing import IO from comfy.comfy_types.node_typing import IO
from comfy.weight_adapter import WeightAdapterBase, WeightAdapterTrainBase, adapters from comfy.weight_adapter import adapters
class TrainSampler(comfy.samplers.Sampler): class TrainSampler(comfy.samplers.Sampler):
@ -375,7 +374,7 @@ class TrainLoraNode:
) )
if existing_adapter is not None: if existing_adapter is not None:
break break
if existing_adapter is not None: if existing_adapter is not None:
train_adapter = existing_adapter.to_train() train_adapter = existing_adapter.to_train()
for name, parameter in train_adapter.named_parameters(): for name, parameter in train_adapter.named_parameters():