Update extract_feature_print.py
Browse files- extract_feature_print.py +1 -1
extract_feature_print.py
CHANGED
|
@@ -190,7 +190,7 @@ os.makedirs(outPath, exist_ok=True)
|
|
| 190 |
def readwave(wav_path, normalize=False):
|
| 191 |
wav, sr = sf.read(wav_path)
|
| 192 |
assert sr == 16000
|
| 193 |
-
if Custom_Embed:
|
| 194 |
feats = torch.from_numpy(wav).float()
|
| 195 |
else:
|
| 196 |
feats = torch.from_numpy(load_audio(wav, sr)).to(dtype).to(device)
|
|
|
|
| 190 |
def readwave(wav_path, normalize=False):
|
| 191 |
wav, sr = sf.read(wav_path)
|
| 192 |
assert sr == 16000
|
| 193 |
+
if Custom_Embed == False:
|
| 194 |
feats = torch.from_numpy(wav).float()
|
| 195 |
else:
|
| 196 |
feats = torch.from_numpy(load_audio(wav, sr)).to(dtype).to(device)
|