Fix Use device_map for transformer models (#124) (58656c3) Improve calculation of num_cols and num_rows (#126) (a45d146)