34 lines
682 B
Markdown
34 lines
682 B
Markdown
---
|
|
language:
|
|
- en
|
|
pipeline_tag: text-generation
|
|
license: cc-by-nc-4.0
|
|
---
|
|
|
|
# AISquare-Instruct-SOLAR-10.7b-v0.5.31
|
|
|
|
## Model Details
|
|
|
|
**Developed by**
|
|
[Inswave Systems](https://www.inswave.com) UI Platform Team
|
|
|
|
**Base Model**
|
|
[upstage/SOLAR-10.7B-v1.0](https://huggingface.co/upstage/SOLAR-10.7B-v1.0)
|
|
|
|
|
|
# Implementation Code
|
|
```python
|
|
from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
import torch
|
|
|
|
repo = "kimwooglae/AISquare-Instruct-SOLAR-10.7b-v0.5.31"
|
|
model = AutoModelForCausalLM.from_pretrained(
|
|
repo,
|
|
return_dict=True,
|
|
torch_dtype=torch.float16,
|
|
device_map='auto'
|
|
)
|
|
tokenizer = AutoTokenizer.from_pretrained(repo)
|
|
```
|
|
|
|
--- |