rohan-bansode commited on
Commit
4535a7d
·
verified ·
1 Parent(s): 177e4d6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -8
README.md CHANGED
@@ -24,16 +24,19 @@ datasets:
24
  - gbharti/finance-alpaca
25
  ---
26
 
27
- 🚀 Quick Start
28
- Install
 
 
29
  pip install mlx-lm
30
- Run Inference
 
 
 
31
  from mlx_lm import load, stream_generate
32
 
33
- # Load model
34
  model, tokenizer = load("rohan-bansode/Qwen-2.5-3B-QLORA")
35
 
36
- # Input prompt
37
  prompt = """
38
  Analyze:
39
  Revenue = 500k
@@ -43,13 +46,12 @@ Rent = 50k
43
  Calculate EBITDAR and explain.
44
  """
45
 
46
- # Stop tokens
47
  stop_sequences = ["<|endoftext|>", "<|im_end|>", "Human:", "Assistant:"]
48
 
49
  print("--- Financial Sandbox Output ---")
50
 
51
- # Generate response
52
  for response in stream_generate(model, tokenizer, prompt, max_tokens=512):
53
  if any(stop in response.text for stop in stop_sequences):
54
  break
55
- print(response.text, end="", flush=True)
 
 
24
  - gbharti/finance-alpaca
25
  ---
26
 
27
+ ## 🚀 Quick Start
28
+
29
+ ### Install
30
+ ```bash
31
  pip install mlx-lm
32
+ ```
33
+
34
+ ### Run Inference
35
+ ```python
36
  from mlx_lm import load, stream_generate
37
 
 
38
  model, tokenizer = load("rohan-bansode/Qwen-2.5-3B-QLORA")
39
 
 
40
  prompt = """
41
  Analyze:
42
  Revenue = 500k
 
46
  Calculate EBITDAR and explain.
47
  """
48
 
 
49
  stop_sequences = ["<|endoftext|>", "<|im_end|>", "Human:", "Assistant:"]
50
 
51
  print("--- Financial Sandbox Output ---")
52
 
 
53
  for response in stream_generate(model, tokenizer, prompt, max_tokens=512):
54
  if any(stop in response.text for stop in stop_sequences):
55
  break
56
+ print(response.text, end="", flush=True)
57
+ ```