kye
/

Andromeda

Model card Files Files and versions

Andromeda / example.py

kye's picture

Upload 73 files

ca4fc4d almost 3 years ago

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)