Downloads · 30 days
10
0% of all-time downloads
zomd/AISquare-Instruct-yi-ko-6b-v0.9.30
AISquare-Instruct-yi-ko-6b-v0.9.30 is a text generation model from zomd. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as cc-by-nc-4.0.
Developed by Inswave Systems UI Platform Team
Downloads · 30 days
10
0% of all-time downloads
All-time downloads
61.1K
Public
Parameters
6.2B
12.4 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors12.4 GB · 100%
From the Hugging Face model README
Developed by Inswave Systems UI Platform Team
Method
Using DPO method and SFT method
Hardware
We utilized an A100x4 * 1 for training our model
Base Model beomi/Yi-Ko-6B
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
repo = "zomd/AISquare-Instruct-yi-ko-6b-v0.9.30"
model = AutoModelForCausalLM.from_pretrained(
repo,
return_dict=True,
torch_dtype=torch.float16,
device_map='auto'
)
tokenizer = AutoTokenizer.from_pretrained(repo)