Upload make_zh.py
Browse files- samples/make_zh.py +1 -1
samples/make_zh.py
CHANGED
@@ -66,7 +66,7 @@ def speed_callable(len_ps):
|
|
66 |
speed = 1 - (len_ps - 83) / 500
|
67 |
return speed * 1.1
|
68 |
|
69 |
-
|
70 |
zh_pipeline = KPipeline(lang_code='z', repo_id=REPO_ID, model=model, en_callable=en_callable)
|
71 |
|
72 |
path = Path(__file__).parent
|
|
|
66 |
speed = 1 - (len_ps - 83) / 500
|
67 |
return speed * 1.1
|
68 |
|
69 |
+
model = KModel(repo_id=REPO_ID).to(device).eval()
|
70 |
zh_pipeline = KPipeline(lang_code='z', repo_id=REPO_ID, model=model, en_callable=en_callable)
|
71 |
|
72 |
path = Path(__file__).parent
|