vikramlingam commited on
Commit
c5827c5
·
verified ·
1 Parent(s): adbc6be

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +23 -0
README.md CHANGED
@@ -74,4 +74,27 @@ prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_
74
  inputs = tokenizer(prompt, return_tensors="pt")
75
  outputs = model.generate(**inputs, max_new_tokens=256, do_sample=False)
76
  print(tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
77
  ```
 
74
  inputs = tokenizer(prompt, return_tensors="pt")
75
  outputs = model.generate(**inputs, max_new_tokens=256, do_sample=False)
76
  print(tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))
77
+ ### 3. With llama.cpp / GGUF (Q4_K_M)
78
+
79
+ ```python
80
+ from llama_cpp import Llama
81
+
82
+ # Download SmolLM2-360M-Post-ASR-Q4_K_M.gguf from this repo
83
+ llm = Llama.from_pretrained(
84
+ repo_id="vikramlingam/SmolLM2-360M-Post-ASR-Engine",
85
+ filename="SmolLM2-360M-Post-ASR-Q4_K_M.gguf",
86
+ n_ctx=1024,
87
+ n_gpu_layers=-1,
88
+ verbose=False,
89
+ )
90
+
91
+ prompt = """<|im_start|>system
92
+ You are a concise, neutral post-ASR formatting engine. Output clean formatting strictly following the task prefix token. Do not include introductory or concluding conversational filler, AI buzzwords, or em dashes.<|im_end|>
93
+ <|im_start|>user
94
+ <|task:format_email|> hey mark we need to shift standup to eleven am please check slide four<|im_end|>
95
+ <|im_start|>assistant
96
+ """
97
+
98
+ output = llm(prompt, max_tokens=256, stop=["<|im_end|>"], temperature=0.0)
99
+ print(output["choices"][0]["text"].strip())
100
  ```