kimwooglae's picture
Create README.md
a8ef130 verified
metadata
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 UI Platform Team

Base Model upstage/SOLAR-10.7B-v1.0

Implementation Code

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)