Spaces:

Cidaut
/

DeMoE

Running

DeMoE / check_file.py

basic functionality

034f4b8 2 months ago

135 Bytes

	import torch

	pt_dict = torch.load('DeMoE.pt', map_location='cpu')
	print(pt_dict['params'].keys())
	print(len(pt_dict['params'].keys()))