Upload folder using huggingface_hub
Browse files
README.md
CHANGED
|
@@ -26,6 +26,7 @@ vllm serve yujiepan/kimi-linear-tiny-random --trust-remote-code
|
|
| 26 |
# tested on transformers==4.57.1
|
| 27 |
import torch
|
| 28 |
import transformers
|
|
|
|
| 29 |
|
| 30 |
model_id = "yujiepan/kimi-linear-tiny-random"
|
| 31 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
|
| 26 |
# tested on transformers==4.57.1
|
| 27 |
import torch
|
| 28 |
import transformers
|
| 29 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 30 |
|
| 31 |
model_id = "yujiepan/kimi-linear-tiny-random"
|
| 32 |
model = AutoModelForCausalLM.from_pretrained(
|