Update README.md
Browse files
README.md
CHANGED
|
@@ -11,10 +11,10 @@ base_model:
|
|
| 11 |
|
| 12 |
<div align="center">
|
| 13 |
|
| 14 |
-
|
| 15 |
<img src="https://cdn-uploads.huggingface.co/production/uploads/650add6348983c90ab688b6e/zT-y1dEjenfRwGk0ezjGc.png" width="300" style="border-radius: 20px;"/>
|
| 16 |
|
| 17 |
-
|
| 18 |
<p style="margin-top: 20px;">
|
| 19 |
<a href="https://example.com/report" style="margin: 0 10px;">
|
| 20 |
📄 <strong>Paper</strong>
|
|
@@ -164,11 +164,11 @@ user_prompt = """
|
|
| 164 |
model = AutoModelForCausalLM.from_pretrained("HeAAAAA/Crab")
|
| 165 |
tokenizer = AutoTokenizer.from_pretrained("HeAAAAA/Crab")
|
| 166 |
|
| 167 |
-
|
| 168 |
inputs_prompt = system_prompt + user_prompt
|
| 169 |
inputs = tokenizer(inputs_prompt, return_tensors="pt")
|
| 170 |
|
| 171 |
-
|
| 172 |
outputs = model.generate(
|
| 173 |
**inputs,
|
| 174 |
max_new_tokens=200,
|
|
|
|
| 11 |
|
| 12 |
<div align="center">
|
| 13 |
|
| 14 |
+
|
| 15 |
<img src="https://cdn-uploads.huggingface.co/production/uploads/650add6348983c90ab688b6e/zT-y1dEjenfRwGk0ezjGc.png" width="300" style="border-radius: 20px;"/>
|
| 16 |
|
| 17 |
+
|
| 18 |
<p style="margin-top: 20px;">
|
| 19 |
<a href="https://example.com/report" style="margin: 0 10px;">
|
| 20 |
📄 <strong>Paper</strong>
|
|
|
|
| 164 |
model = AutoModelForCausalLM.from_pretrained("HeAAAAA/Crab")
|
| 165 |
tokenizer = AutoTokenizer.from_pretrained("HeAAAAA/Crab")
|
| 166 |
|
| 167 |
+
|
| 168 |
inputs_prompt = system_prompt + user_prompt
|
| 169 |
inputs = tokenizer(inputs_prompt, return_tensors="pt")
|
| 170 |
|
| 171 |
+
|
| 172 |
outputs = model.generate(
|
| 173 |
**inputs,
|
| 174 |
max_new_tokens=200,
|