fix: Fixed local distillation (MinishLab#149)

Author: Pringled

model2vec/distill/distillation.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import logging
+import os
 import re
 from typing import Literal, Union
 
@@ -141,13 +142,19 @@ def distill_from_model(
         "hidden_dim": embeddings.shape[1],
         "seq_length": 1000000,  # Set this to a high value since we don't have a sequence length limit.
     }
-    # Get the language from the model card
-    try:
-        info = model_info(model_name)
-        language = info.cardData.get("language")
-    except RepositoryNotFoundError:
-        logger.info("No model info found for model. Setting `language` to None.")
+
+    if os.path.exists(model_name):
+        # Using a local model. Get the model name from the path.
+        model_name = os.path.basename(model_name)
         language = None
+    else:
+        # Get the language from the model card.
+        try:
+            info = model_info(model_name)
+            language = info.cardData.get("language", None)
+        except RepositoryNotFoundError:
+            logger.info("No model info found for the model. Setting language to None.")
+            language = None
 
     return StaticModel(
         vectors=embeddings, tokenizer=new_tokenizer, config=config, base_model_name=model_name, language=language