298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-19 19:03:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix typing syntax error

Browse Source
This commit is contained in:
Kohaku-Blueleaf 2025-04-08 18:46:53 +08:00
parent 726fdfcaa0
commit a220e5ca80
3 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/weight_adapter/glora.py
View File

@ -1,5 +1,6 @@
import torch from typing import Optional
import torch
from .base import WeightAdapterBase from .base import WeightAdapterBase
@ -18,7 +19,7 @@ class GLoRAAdapter(WeightAdapterBase):
alpha: float, alpha: float,
dora_scale: torch.Tensor, dora_scale: torch.Tensor,
loaded_keys: set[str] = None, loaded_keys: set[str] = None,
) -> "GLoRAAdapter" | None: ) -> Optional["GLoRAAdapter"]:
if loaded_keys is None: if loaded_keys is None:
loaded_keys = set() loaded_keys = set()
a1_name = "{}.a1.weight".format(x) a1_name = "{}.a1.weight".format(x)

5
comfy/weight_adapter/loha.py
View File

@ -1,5 +1,6 @@
import torch from typing import Optional
import torch
from .base import WeightAdapterBase from .base import WeightAdapterBase
@ -18,7 +19,7 @@ class LoHaAdapter(WeightAdapterBase):
alpha: float, alpha: float,
dora_scale: torch.Tensor, dora_scale: torch.Tensor,
loaded_keys: set[str] = None, loaded_keys: set[str] = None,
) -> "LoHaAdapter" | None: ) -> Optional["LoHaAdapter"]:
if loaded_keys is None: if loaded_keys is None:
loaded_keys = set() loaded_keys = set()

5
comfy/weight_adapter/lokr.py
View File

@ -1,5 +1,6 @@
import torch from typing import Optional
import torch
from .base import WeightAdapterBase from .base import WeightAdapterBase
@ -18,7 +19,7 @@ class LoKrAdapter(WeightAdapterBase):
alpha: float, alpha: float,
dora_scale: torch.Tensor, dora_scale: torch.Tensor,
loaded_keys: set[str] = None, loaded_keys: set[str] = None,
) -> "LoKrAdapter" | None: ) -> Optional["LoKrAdapter"]:
if loaded_keys is None: if loaded_keys is None:
loaded_keys = set() loaded_keys = set()
lokr_w1_name = "{}.lokr_w1".format(x) lokr_w1_name = "{}.lokr_w1".format(x)