kye
/

Andromeda

Model card Files Files and versions Community

Andromeda / example.py

kye's picture

kye

Upload 73 files

ca4fc4d over 1 year 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)