From f513f0b2e0d072325082e189a21f7d5d16168cbd Mon Sep 17 00:00:00 2001
From: Zzzyt <37258146+Zzzzzzyt@users.noreply.github.com>
Date: Tue, 30 Jan 2024 11:51:40 +0800
Subject: [PATCH] Fix half argument (#1683) (#1775)

* for GTX1650

* testing torch-profiler

* no more profiler & change default audio

* longer slice

* fluid container

* cache rmvpe and ui tweaks

* get my changes back after merge

* format code

* only load rmvpe when necessary

* fix rmvpe & config bug

* fix is_half again

* manual sync with upstream

* fix bool() for is_half

* revert changes for PR

* revert again

* typo

* match lower case
---
 infer/modules/train/extract_feature_print.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/infer/modules/train/extract_feature_print.py b/infer/modules/train/extract_feature_print.py
index f7c89c7..96a69de 100644
--- a/infer/modules/train/extract_feature_print.py
+++ b/infer/modules/train/extract_feature_print.py
@@ -11,13 +11,13 @@ i_part = int(sys.argv[3])
 if len(sys.argv) == 7:
     exp_dir = sys.argv[4]
     version = sys.argv[5]
-    is_half = bool(sys.argv[6])
+    is_half = sys.argv[6].lower() == "true"
 else:
     i_gpu = sys.argv[4]
     exp_dir = sys.argv[5]
     os.environ["CUDA_VISIBLE_DEVICES"] = str(i_gpu)
     version = sys.argv[6]
-    is_half = bool(sys.argv[7])
+    is_half = sys.argv[7].lower() == "true"
 import fairseq
 import numpy as np
 import soundfile as sf