support torch_br (#20)

DeepLink-org · Sep 3, 2024 · d2c2494 · d2c2494
1 parent 53b0eea
commit d2c2494
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/ditorch/__init__.py b/ditorch/__init__.py
@@ -21,4 +21,11 @@
 except:
     pass
 
+try:
+    from ditorch import torch_biren_adapter
+
+    framework = "torch_br:" + torch_biren_adapter.torch_br.__version__
+except:
+    pass
+
 print(f"ditorch.framework: {framework}")
diff --git a/ditorch/torch_biren_adapter.py b/ditorch/torch_biren_adapter.py
@@ -0,0 +1,3 @@
+# Copyright (c) 2024, DeepLink.
+import torch_br
+from torch_br.contrib import transfer_to_supa