AlBERTmini

calogero-jerik-scozzaro commited on about 1 month ago

Commit

8efff21

verified ·

1 Parent(s): bdcc3fe

Update modeling_matformer.py

Files changed (1) hide show

modeling_matformer.py CHANGED Viewed

@@ -26,7 +26,7 @@ except ImportError as e:
     try:
         subprocess.check_call([
             sys.executable, "-m", "pip", "install",
-            "git+https://github.com/mrinaldi97/matformer.git@6762b2e00ec8b76bb1fb561580c895f58a51ca55"
         ])
         from matformer.huggingface_integration import (
@@ -41,7 +41,9 @@ except ImportError as e:
     except Exception as install_error:
         raise ImportError(
-            "Failed to install Matformer. Install manually:\n"
-            "  pip install git+https://github.com/mrinaldi97/matformer.git\n"
-            "Or set MATFORMER_ROOT environment variable"
         ) from install_error

     try:
         subprocess.check_call([
             sys.executable, "-m", "pip", "install",
+            "git+https://github.com/mrinaldi97/matformer.git@alberturin-v1",
         ])
         from matformer.huggingface_integration import (
     except Exception as install_error:
         raise ImportError(
+            "Failed to install Matformer automatically.\n"
+            "Install the AlBERTurin-compatible version manually with:\n\n"
+            "  pip install "
+            "git+https://github.com/mrinaldi97/matformer.git@alberturin-v1\n\n"
+            "Alternatively, set the MATFORMER_ROOT environment variable."
         ) from install_error