From a220e5ca80d1f886c241a3af28be84e92e762f6f Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Tue, 8 Apr 2025 18:46:53 +0800
Subject: [PATCH] Fix typing syntax error

---
 comfy/weight_adapter/glora.py | 5 +++--
 comfy/weight_adapter/loha.py  | 5 +++--
 comfy/weight_adapter/lokr.py  | 5 +++--
 3 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/comfy/weight_adapter/glora.py b/comfy/weight_adapter/glora.py
index 89574f9f..206fc6e2 100644
--- a/comfy/weight_adapter/glora.py
+++ b/comfy/weight_adapter/glora.py
@@ -1,5 +1,6 @@
-import torch
+from typing import Optional
 
+import torch
 from .base import WeightAdapterBase
 
 
@@ -18,7 +19,7 @@ class GLoRAAdapter(WeightAdapterBase):
         alpha: float,
         dora_scale: torch.Tensor,
         loaded_keys: set[str] = None,
-    ) -> "GLoRAAdapter" | None:
+    ) -> Optional["GLoRAAdapter"]:
         if loaded_keys is None:
             loaded_keys = set()
         a1_name = "{}.a1.weight".format(x)
diff --git a/comfy/weight_adapter/loha.py b/comfy/weight_adapter/loha.py
index c0303925..7a6cab24 100644
--- a/comfy/weight_adapter/loha.py
+++ b/comfy/weight_adapter/loha.py
@@ -1,5 +1,6 @@
-import torch
+from typing import Optional
 
+import torch
 from .base import WeightAdapterBase
 
 
@@ -18,7 +19,7 @@ class LoHaAdapter(WeightAdapterBase):
         alpha: float,
         dora_scale: torch.Tensor,
         loaded_keys: set[str] = None,
-    ) -> "LoHaAdapter" | None:
+    ) -> Optional["LoHaAdapter"]:
         if loaded_keys is None:
             loaded_keys = set()
 
diff --git a/comfy/weight_adapter/lokr.py b/comfy/weight_adapter/lokr.py
index 85bad7ec..da1b1551 100644
--- a/comfy/weight_adapter/lokr.py
+++ b/comfy/weight_adapter/lokr.py
@@ -1,5 +1,6 @@
-import torch
+from typing import Optional
 
+import torch
 from .base import WeightAdapterBase
 
 
@@ -18,7 +19,7 @@ class LoKrAdapter(WeightAdapterBase):
         alpha: float,
         dora_scale: torch.Tensor,
         loaded_keys: set[str] = None,
-    ) -> "LoKrAdapter" | None:
+    ) -> Optional["LoKrAdapter"]:
         if loaded_keys is None:
             loaded_keys = set()
         lokr_w1_name = "{}.lokr_w1".format(x)