SupraLabs/Supra-50M-Instruct

#4
.gitattributes CHANGED
@@ -34,4 +34,3 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  supra-50m-instruct-f16.gguf filter=lfs diff=lfs merge=lfs -text
37
- Supra-50M-f16.gguf filter=lfs diff=lfs merge=lfs -text
 
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  supra-50m-instruct-f16.gguf filter=lfs diff=lfs merge=lfs -text
 
README.md CHANGED
@@ -20,7 +20,6 @@ tags:
20
  - slm
21
  base_model:
22
  - SupraLabs/Supra-50M-Base
23
- new_version: SupraLabs/Supra-1.5-50M-Instruct-exp
24
  ---
25
 
26
  # πŸ¦… Supra-50M Instruct
@@ -78,10 +77,8 @@ For more details, the full code, configs and weights, please refer to [https://h
78
  ```python
79
  import os
80
  import warnings
81
- import time
82
 
83
  os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
84
- os.environ["CUDA_VISIBLE_DEVICES"] = "0"
85
  warnings.filterwarnings("ignore", category=UserWarning, module="transformers")
86
 
87
  import torch
@@ -89,13 +86,6 @@ from transformers import pipeline, AutoTokenizer, logging
89
 
90
  logging.set_verbosity_error()
91
 
92
- # ── Global variables ──────────────────────────────────────────────────────────
93
-
94
- end = time.time()
95
- start = time.time()
96
- tokens = []
97
-
98
-
99
  # ── Config ────────────────────────────────────────────────────────────────────
100
 
101
  MODEL_ID = "SupraLabs/Supra-50M-Instruct"
@@ -140,7 +130,7 @@ def build_prompt(instruction: str, input_text: str = "") -> str:
140
 
141
  def generate(instruction: str, input_text: str = "", max_new_tokens: int = MAX_NEW_TOKENS) -> str:
142
  prompt = build_prompt(instruction, input_text)
143
- start = time.time()
144
  result = pipe(
145
  prompt,
146
  max_new_tokens=max_new_tokens,
@@ -154,13 +144,7 @@ def generate(instruction: str, input_text: str = "", max_new_tokens: int = MAX_N
154
  return_full_text=False,
155
  generation_config=None
156
  )
157
- end = time.time()
158
-
159
- generated_text = result[0]["generated_text"]
160
- tokens = pipe.tokenizer(generated_text)["input_ids"]
161
 
162
- return generated_text, tokens, end, start
163
-
164
  # ── Interactive loop ──────────────────────────────────────────────────────────
165
 
166
  if __name__ == "__main__":
@@ -174,13 +158,8 @@ if __name__ == "__main__":
174
  inp = input("Input (optional, Enter to skip): ").strip()
175
 
176
  print("-" * 50)
177
- text, tokens, end, start = generate(instruction, inp)
178
- print(text)
179
  print()
180
-
181
- print(f"Generated tokens: {len(tokens)}")
182
- print(f"Time: {end - start:.2f}s")
183
- print(f"Speed: {len(tokens) / (end - start):.2f} tokens/sec")
184
  ```
185
 
186
  ---
 
20
  - slm
21
  base_model:
22
  - SupraLabs/Supra-50M-Base
 
23
  ---
24
 
25
  # πŸ¦… Supra-50M Instruct
 
77
  ```python
78
  import os
79
  import warnings
 
80
 
81
  os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
 
82
  warnings.filterwarnings("ignore", category=UserWarning, module="transformers")
83
 
84
  import torch
 
86
 
87
  logging.set_verbosity_error()
88
 
 
 
 
 
 
 
 
89
  # ── Config ────────────────────────────────────────────────────────────────────
90
 
91
  MODEL_ID = "SupraLabs/Supra-50M-Instruct"
 
130
 
131
  def generate(instruction: str, input_text: str = "", max_new_tokens: int = MAX_NEW_TOKENS) -> str:
132
  prompt = build_prompt(instruction, input_text)
133
+
134
  result = pipe(
135
  prompt,
136
  max_new_tokens=max_new_tokens,
 
144
  return_full_text=False,
145
  generation_config=None
146
  )
 
 
 
 
147
 
 
 
148
  # ── Interactive loop ──────────────────────────────────────────────────────────
149
 
150
  if __name__ == "__main__":
 
158
  inp = input("Input (optional, Enter to skip): ").strip()
159
 
160
  print("-" * 50)
161
+ print(generate(instruction, inp))
 
162
  print()
 
 
 
 
163
  ```
164
 
165
  ---
Supra-50M-f16.gguf β†’ supra-50m-instruct-f16.gguf RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:14a5840b3e66cd159eae57675dab4ec4e8eb074b823ee68c2f65eb58a655f6f7
3
- size 104739616
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e104f91e7aeed5c11d1601461b93ddc6b71c8d607d064c3afd8d0bfe5590647f
3
+ size 104741056