From 9dec17feafd8ec4efb0780b28d6bb7e3a79b2468 Mon Sep 17 00:00:00 2001
From: McCloudS <64094529+McCloudS@users.noreply.github.com>
Date: Sat, 10 Feb 2024 17:36:24 -0700
Subject: [PATCH] Fixed me handling ints and bools wrong in env variables.

---
 subgen/subgen.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/subgen/subgen.py b/subgen/subgen.py
index 2cdd311..8db9e8c 100644
--- a/subgen/subgen.py
+++ b/subgen/subgen.py
@@ -50,9 +50,9 @@ model_location = os.getenv('MODEL_PATH', '.')
 transcribe_folders = os.getenv('TRANSCRIBE_FOLDERS', '')
 transcribe_or_translate = os.getenv('TRANSCRIBE_OR_TRANSLATE', 'translate')
 force_detected_language_to = os.getenv('FORCE_DETECTED_LANGUAGE_TO', '')
-hf_transformers = os.getenv('HF_TRANSFORMERS', False)
+hf_transformers = convert_to_bool(os.getenv('HF_TRANSFORMERS', False))
 hf_batch_size = int(os.getenv('HF_BATCH_SIZE', 24))
-clear_vram_on_complete = os.getenv('CLEAR_VRAM_ON_COMPLETE', True)
+clear_vram_on_complete = convert_to_bool(os.getenv('CLEAR_VRAM_ON_COMPLETE', True))
 compute_type = os.getenv('COMPUTE_TYPE', 'auto')
 if transcribe_device == "gpu":
     transcribe_device = "cuda"