Andromeda
0.0.11
Selamat datang di Andromeda, Model Bahasa Tercepat, Paling Kreatif, dan Andal yang Pernah Dibuat, latih versi Anda sendiri, lakukan inferensi, dan sempurnakan versi Anda sendiri dengan skrip plug-in dan play sederhana, mulai dalam 10 detik:
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 )
Tetapkan variabel lingkungan:
ENTITY_NAME
: Nama proyek tongkat sihir AndaOUTPUT_DIR
: Direktori untuk menyimpan bobot (misal, ./weights
)MASTER_ADDR
: Untuk pelatihan terdistribusiMASTER_PORT
Untuk pelatihan terdistribusi port masterRANK
- Jumlah layanan nodeWORLD_SIZE
Jumlah GPUKonfigurasikan pelatihan:
Untuk informasi lebih lanjut, lihat SOP Pelatihan.
Lisensi Apache