Andromeda
0.0.11
ยินดีต้อนรับสู่ Andromeda โมเดลภาษาที่เร็วที่สุด สร้างสรรค์ที่สุด และเชื่อถือได้เท่าที่เคยมีมา ฝึกฝนรูปแบบของคุณเอง ดำเนินการอนุมาน และปรับแต่งรูปแบบของคุณเองด้วยสคริปต์ Plug-in และ Play ที่เรียบง่าย เริ่มต้นได้ใน 10 วินาที:
python3.11 -m pip install --upgrade andromeda-torch
import torch
from andromeda . configs import Andromeda1Billion
model = Andromeda1Billion ()
x = torch . randint ( 0 , 256 , ( 1 , 1024 )). cuda ()
out = model ( x ) # (1, 1024, 20000)
print ( out )
from andromeda_torch import Tokenizer
from andromeda_torch . configs import Andromeda1Billion
model = Andromeda1Billion ()
tokenizer = Tokenizer ()
encoded_text = tokenizer . encode ( "Hello world!" )
out = model ( encoded_text )
print ( out )
ตั้งค่าตัวแปรสภาพแวดล้อม:
ENTITY_NAME
: ชื่อโครงการไม้กายสิทธิ์ของคุณOUTPUT_DIR
: ไดเรกทอรีสำหรับบันทึกน้ำหนัก (เช่น ./weights
weights )MASTER_ADDR
: สำหรับการฝึกอบรมแบบกระจายMASTER_PORT
สำหรับการฝึกอบรมแบบกระจายพอร์ตหลักRANK
- จำนวนบริการโหนดWORLD_SIZE
จำนวน GPUกำหนดค่าการฝึกอบรม:
สำหรับข้อมูลเพิ่มเติม โปรดดู SOP การฝึกอบรม
ใบอนุญาต Apache