DH-5746/fixing the seed to avoid randomness

Dataherald · Apr 18, 2024 · b92a6d2 · b92a6d2
1 parent 828c64d
commit b92a6d2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dataherald/model/chat_model.py b/dataherald/model/chat_model.py
@@ -27,6 +27,7 @@ def get_model(
                 model_name=model_name,
                 openai_api_key=api_key,
                 openai_api_base=api_base,
+                seed=0,
                 **kwargs
             )
         if model_family == "anthropic":