More precise condition

1 year ago · 59d7db9aad
parent adc599b0a6
commit 59d7db9aad
1 changed files with 1 additions and 1 deletions
--- a/gpt4all-bindings/python/gpt4all/gpt4all.py
+++ b/gpt4all-bindings/python/gpt4all/gpt4all.py
@ -73,7 +73,7 @@ class GPT4All():
        """
        
        model_filename = model_name
-        if ".bin" not in model_filename:
+        if not model_filename.endswith(".bin"):
            model_filename += ".bin"

        # Validate download directory