Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -71,9 +71,9 @@ max_speech_ms=float("inf")
|
|
71 |
# "distil-whisper/distil-large-v3",
|
72 |
# torch_dtype="float16",
|
73 |
# ).to("cpu")
|
74 |
-
LM_tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM-
|
75 |
LM_model = AutoModelForCausalLM.from_pretrained(
|
76 |
-
"HuggingFaceTB/SmolLM-
|
77 |
).to("cpu")
|
78 |
LM_pipe = pipeline(
|
79 |
"text-generation", model=LM_model, tokenizer=LM_tokenizer, device="cpu"
|
|
|
71 |
# "distil-whisper/distil-large-v3",
|
72 |
# torch_dtype="float16",
|
73 |
# ).to("cpu")
|
74 |
+
LM_tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM-360M-Instruct")
|
75 |
LM_model = AutoModelForCausalLM.from_pretrained(
|
76 |
+
"HuggingFaceTB/SmolLM-360M-Instruct", torch_dtype="float16", trust_remote_code=True
|
77 |
).to("cpu")
|
78 |
LM_pipe = pipeline(
|
79 |
"text-generation", model=LM_model, tokenizer=LM_tokenizer, device="cpu"
|