kye
/

Andromeda

Model card Files Files and versions

xet

Community

Andromeda / example.py

kye

Upload 73 files

ca4fc4d about 3 years ago

Raw History Blame Contribute Delete

171 Bytes

import torch
from Andromeda.configs import Andromeda1Billion

model =  Andromeda1Billion().cuda()

x = torch.randint(0, 256, (1, 1024)).cuda()

model(x) # (1, 1024, 20000)

	import torch
	from Andromeda.configs import Andromeda1Billion

	model = Andromeda1Billion().cuda()

	x = torch.randint(0, 256, (1, 1024)).cuda()

	model(x) # (1, 1024, 20000)