From cf36ce977c0b8bd54c906e63ce31931ac060178f Mon Sep 17 00:00:00 2001
From: anyvoice <130631963+anyvoiceai@users.noreply.github.com>
Date: 星期日, 02 七月 2023 09:12:18 +0800
Subject: [PATCH] Creates tensor in target device to avoid high CPU occupation. (#695)
---
tests/test_sv_inference_pipeline.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/tests/test_sv_inference_pipeline.py b/tests/test_sv_inference_pipeline.py
index 60ece2d..c4e427e 100644
--- a/tests/test_sv_inference_pipeline.py
+++ b/tests/test_sv_inference_pipeline.py
@@ -35,4 +35,4 @@
logger.info(f"Similarity {rec_result['scores']}")
if __name__ == '__main__':
- unittest.main()
+ unittest.main()
\ No newline at end of file
--
Gitblit v1.9.1