LH-Tech-AI commited on
Commit
e73f003
Β·
verified Β·
1 Parent(s): 39d052b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -4
README.md CHANGED
@@ -39,19 +39,27 @@ For more details, the full code, configs and weights, please refer to [https://h
39
  ## πŸš€ Inference
40
 
41
  ```python
 
 
 
 
 
 
42
  import torch
43
- from transformers import pipeline, AutoTokenizer
 
 
44
 
45
  # ── Config ────────────────────────────────────────────────────────────────────
46
 
47
- MODEL_ID = "SupraLabs/Supra-50M-Instruct/"
48
  MAX_NEW_TOKENS = 512
49
 
50
  # ── Load pipeline directly from HF ────────────────────────────────────────────
51
 
52
  print(f"[*] Loading SFT model and tokenizer from HF Hub ({MODEL_ID})...")
53
 
54
- tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
55
 
56
  pipe = pipeline(
57
  "text-generation",
@@ -97,7 +105,8 @@ def generate(instruction: str, input_text: str = "", max_new_tokens: int = MAX_N
97
  repetition_penalty=1.15,
98
  pad_token_id=pipe.tokenizer.pad_token_id,
99
  eos_token_id=pipe.tokenizer.eos_token_id,
100
- return_full_text=False
 
101
  )
102
 
103
  return result[0]['generated_text'].strip()
 
39
  ## πŸš€ Inference
40
 
41
  ```python
42
+ import os
43
+ import warnings
44
+
45
+ os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
46
+ warnings.filterwarnings("ignore", category=UserWarning, module="transformers")
47
+
48
  import torch
49
+ from transformers import pipeline, AutoTokenizer, logging
50
+
51
+ logging.set_verbosity_error()
52
 
53
  # ── Config ────────────────────────────────────────────────────────────────────
54
 
55
+ MODEL_ID = "SupraLabs/Supra-50M-Instruct"
56
  MAX_NEW_TOKENS = 512
57
 
58
  # ── Load pipeline directly from HF ────────────────────────────────────────────
59
 
60
  print(f"[*] Loading SFT model and tokenizer from HF Hub ({MODEL_ID})...")
61
 
62
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, clean_up_tokenization_spaces=False)
63
 
64
  pipe = pipeline(
65
  "text-generation",
 
105
  repetition_penalty=1.15,
106
  pad_token_id=pipe.tokenizer.pad_token_id,
107
  eos_token_id=pipe.tokenizer.eos_token_id,
108
+ return_full_text=False,
109
+ generation_config=None
110
  )
111
 
112
  return result[0]['generated_text'].strip()