support custom svd (#536)

siliconflow · Jan 27, 2024 · 842c480 · 842c480
1 parent f8484d1
commit 842c480
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/onediff/infer_compiler/transform/builtin_transform.py b/src/onediff/infer_compiler/transform/builtin_transform.py
@@ -105,7 +105,7 @@ def __getattribute__(self, attribute):
         ):
             return flow.Generator()
         elif (
-            isinstance(self._oflow_proxy_submod, (torch.nn.Conv2d, torch.nn.Conv3d))
+            isinstance(self._oflow_proxy_submod, (torch.nn.Conv1d, torch.nn.Conv2d, torch.nn.Conv3d))
             and attribute == "channel_pos"
         ):
             return "channels_first"