zwpride commited on
Commit
b87cf3a
·
verified ·
1 Parent(s): c18e85b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -36
README.md CHANGED
@@ -76,42 +76,6 @@ vllm serve $MODEL --port 8080 \
76
  --cudagraph-capture-sizes 1 2 4 8 12 16 24 32
77
  ```
78
 
79
- ```python
80
- import torch
81
- from transformers import AutoModelForCausalLM, AutoTokenizer
82
-
83
- repo_id = "Multilingual-Multimodal-NLP/LoopCoder-V2"
84
-
85
- tokenizer = AutoTokenizer.from_pretrained(repo_id, trust_remote_code=True)
86
- model = AutoModelForCausalLM.from_pretrained(
87
- repo_id,
88
- torch_dtype=torch.bfloat16,
89
- device_map="auto",
90
- trust_remote_code=True,
91
- )
92
-
93
- messages = [
94
- {"role": "user", "content": "Write a Python function that checks whether a string is a palindrome."}
95
- ]
96
-
97
- inputs = tokenizer.apply_chat_template(
98
- messages,
99
- add_generation_prompt=True,
100
- tokenize=True,
101
- return_tensors="pt",
102
- ).to(model.device)
103
-
104
- outputs = model.generate(
105
- inputs,
106
- max_new_tokens=512,
107
- do_sample=True,
108
- temperature=0.6,
109
- top_p=0.95,
110
- top_k=20,
111
- )
112
-
113
- print(tokenizer.decode(outputs[0][inputs.shape[-1]:], skip_special_tokens=True))
114
- ```
115
 
116
  ## Training Data
117
 
 
76
  --cudagraph-capture-sizes 1 2 4 8 12 16 24 32
77
  ```
78
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
79
 
80
  ## Training Data
81