More precise condition

This commit is contained in:
Konstantin Gukov 2023-05-25 17:17:00 +02:00 committed by Richard Guo
parent adc599b0a6
commit 59d7db9aad

View File

@ -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