Merge branch 'main' into sunspot-tests

globus-labs · May 23, 2024 · d9f3414 · d9f3414
2 parents 1fd87d6 + f0ace80
commit d9f3414
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/mofa/difflinker_train.py b/mofa/difflinker_train.py
@@ -7,6 +7,12 @@
 from pytorch_lightning.callbacks import TQDMProgressBar
 from pytorch_lightning.strategies import SingleDeviceStrategy
 
+try:
+    import intel_extension_for_pytorch as ipex  # noqa: F401
+    import oneccl_bindings_for_pytorch  # noqa: F401
+except ImportError:
+    pass
+
 try:
     import intel_extension_for_pytorch as ipex  # noqa: F401
     import oneccl_bindings_for_pytorch  # noqa: F401
@@ -151,7 +157,6 @@ def main(
             if '.' in args.train_data_prefix:
                 context_node_nf += 1
 
-
             # Lock XPU to single device for now
             strategy = 'auto'
             if args.device == 'xpu':