minor display fix, enabled GPU for TTS by default

FontaineRiant · Apr 18, 2024 · 3b1a6f9 · 3b1a6f9
1 parent 62f530f
commit 3b1a6f9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/main.py b/main.py
@@ -392,16 +392,16 @@ def loop_choice(self):
             question = [
                 {
                     'type': list_input_type,
-                    'name': 'model_name',
-                    'message': f'choice:',
+                    'name': 'choice',
+                    'message': f'\nchoice:',
                     'choices': choices
                 }
             ]
 
             user_input = {}
             while not user_input:
                 user_input = prompt(question, style=self.style)
-            user_input = user_input['model_name']
+            user_input = user_input['choice']
 
             if user_input == '< menu >':
                 return

diff --git a/play.sh b/play.sh
@@ -1,3 +1,3 @@
 source ./venv/bin/activate
-python main.py --censor --model "KoboldAI/OPT-2.7B-Nerys-v2" --cputts
+python main.py --censor --model "KoboldAI/OPT-2.7B-Nerys-v2"
 read -sn 1 -p "Press any key to continue..."