module.original_name would return a string containing the original module name, e.g.:
lin = nn.Linear(1, 1)
s = torch.jit.script(lin)
print(s.original_name)
> 'Linear'

so you could try to use this attribute instead.