From f56063b54b2d314bd0739b11b346d9e3012be106 Mon Sep 17 00:00:00 2001 From: wunderwuzzi23 Date: Tue, 4 Apr 2023 17:49:35 -0700 Subject: [PATCH] clean up some code, v.2.1 (bug fix) --- yolo.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/yolo.py b/yolo.py index b067879..6911b85 100755 --- a/yolo.py +++ b/yolo.py @@ -56,18 +56,11 @@ def print_usage(): print("Argument: -a: Prompt the user before running the command (only useful when safety is off)") print() - - #Ensure safety switch has a correct value - yolo_safety_switch = bool(config["safety"]) - - # if config["safety"] != True: - # yolo_safety_switch = "off" - print("Current configuration per yolo.yaml:") print("* Model : " + str(config["model"])) print("* Temperature : " + str(config["temperature"])) print("* Max. Tokens : " + str(config["max_tokens"])) - print("* Safety : " + str(yolo_safety_switch)) + print("* Safety : " + str(bool(config["safety"]))) def get_os_friendly_name():