Siddhant commited on
Commit
a3aa488
1 Parent(s): 16c6824

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-135M-Instruct")
75
  LM_model = AutoModelForCausalLM.from_pretrained(
76
- "HuggingFaceTB/SmolLM-135M-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"
 
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"