add back cm torch

fixed cm torch issue