Add set_seed (#17)

* Add sampling from hugging face

* Add set seed function

* Fix merge

* Fix comment

Author: neubig

llments/lm/base/empirical.py

@@ -51,6 +51,9 @@ def calculate_probability(self, x: str) -> float:
         # Implementation logic
         raise NotImplementedError("This is not implemented yet.")
 
+    def set_seed(self, seed: int):
+        random.seed(seed)
+
 
 def load_from_text_file(text_file: str):
     """Load the distribution from a text file."""

llments/lm/base/hugging_face.py

@@ -1,5 +1,5 @@
 from llments.lm.lm import LanguageModel
-from transformers import pipeline
+from transformers import pipeline, set_seed
 
 
 class HuggingFaceLM(LanguageModel):
@@ -64,6 +64,14 @@ def generate(
         )
         return [res["generated_text"] for res in results]
 
+    def set_seed(self, seed: int):
+        """Set the seed for the language model.
+
+        Args:
+            seed: The seed to set for the language model.
+        """
+        set_seed(seed)
+
 
 def load_from_spec(spec_file: str) -> HuggingFaceLM:
     """Load a language model from a specification file.

llments/lm/lm.py

@@ -39,3 +39,12 @@ def generate(
             str: Sampled output sequences from the language model.
         """
         ...
+
+    @abc.abstractmethod
+    def set_seed(self, seed: int):
+        """Set the seed for the language model.
+
+        Args:
+            seed: The seed to set for the language model.
+        """
+        ...