update model used in inference
Browse files- adapter_model.bin +1 -1
- handler.py +2 -1
adapter_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 63056269
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7cd99a2a9834d646d9f6145b3166b3c4a2182792c9d1e4581b608d96c2cb6c1b
|
| 3 |
size 63056269
|
handler.py
CHANGED
|
@@ -12,7 +12,8 @@ from peft import LoraConfig, PeftModel, LoraModel, LoraConfig, get_peft_model, P
|
|
| 12 |
class EndpointHandler():
|
| 13 |
def __init__(self, path=""):
|
| 14 |
# Preload all the elements you are going to need at inference.
|
| 15 |
-
peft_model_id = "cathyi/tw-tw-openai-whisper-large-v2-Lora-epoch5-total5epoch"
|
|
|
|
| 16 |
language = "Chinese"
|
| 17 |
task = "transcribe"
|
| 18 |
peft_config = PeftConfig.from_pretrained(peft_model_id)
|
|
|
|
| 12 |
class EndpointHandler():
|
| 13 |
def __init__(self, path=""):
|
| 14 |
# Preload all the elements you are going to need at inference.
|
| 15 |
+
# peft_model_id = "cathyi/tw-tw-openai-whisper-large-v2-Lora-epoch5-total5epoch"
|
| 16 |
+
peft_model_id = path
|
| 17 |
language = "Chinese"
|
| 18 |
task = "transcribe"
|
| 19 |
peft_config = PeftConfig.from_pretrained(peft_model_id)
|