初始化项目,由ModelHub XC社区提供模型
Model: mhenrichsen/hestenettetLM Source: Original Platform
This commit is contained in:
38
README.md
Normal file
38
README.md
Normal file
@@ -0,0 +1,38 @@
|
||||
---
|
||||
license: mit
|
||||
datasets:
|
||||
- mhenrichsen/hestenettet
|
||||
language:
|
||||
- da
|
||||
---
|
||||
|
||||
# HestenettetLM
|
||||
En dansk LLM trænet på hele hestenettet over 3 epoker.
|
||||
|
||||
Modellen er baseret på Mistral 7b, og har et kontekstvindue på 8k.
|
||||
|
||||
```python
|
||||
from transformers import AutoTokenizer, TextStreamer, AutoModelForCausalLM
|
||||
|
||||
model = AutoModelForCausalLM.from_pretrained("mhenrichsen/hestenettetLM")
|
||||
tokenizer = AutoTokenizer.from_pretrained("mhenrichsen/hestenettetLM")
|
||||
streamer = TextStreamer(tokenizer, skip_special_tokens=True)
|
||||
|
||||
|
||||
tokens = tokenizer(
|
||||
"Den bedste hest er en ",
|
||||
return_tensors='pt'
|
||||
)['input_ids']
|
||||
|
||||
# Generate output
|
||||
generation_output = model.generate(
|
||||
tokens,
|
||||
streamer=streamer,
|
||||
max_length = 8194,
|
||||
)
|
||||
```
|
||||
|
||||
Eksempel:
|
||||
"Den bedste hest er en "
|
||||
bliver til:
|
||||
"Den bedste hest er en veltrænet hest."
|
||||
Reference in New Issue
Block a user