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
|