Minor

models/mask_transformer/transformer.py

@@ -179,9 +179,10 @@ class MaskTransformer(nn.Module):
         clip_model, clip_preprocess = clip.load(clip_version, device='cpu',
                                                 jit=False)  # Must set jit=False for training
         # Cannot run on cpu
-        clip.model.convert_weights(
-            clip_model)  # Actually this line is unnecessary since clip by default already on float16
-        # Date 0707: It's necessary, only unecessary when load directly to gpu. Disable if need to run on cpu
+        if str(self.evice) != "cpu":
+            clip.model.convert_weights(
+                clip_model)  # Actually this line is unnecessary since clip by default already on float16
+            # Date 0707: It's necessary, only unecessary when load directly to gpu. Disable if need to run on cpu
 
         # Freeze CLIP weights
         clip_model.eval()
@@ -731,9 +732,10 @@ class ResidualTransformer(nn.Module):
         clip_model, clip_preprocess = clip.load(clip_version, device='cpu',
                                                 jit=False)  # Must set jit=False for training
         # Cannot run on cpu
-        clip.model.convert_weights(
-            clip_model)  # Actually this line is unnecessary since clip by default already on float16
-        # Date 0707: It's necessary, only unecessary when load directly to gpu. Disable if need to run on cpu
+        if str(self.evice) != "cpu":
+            clip.model.convert_weights(
+                clip_model)  # Actually this line is unnecessary since clip by default already on float16
+            # Date 0707: It's necessary, only unecessary when load directly to gpu. Disable if need to run on cpu
 
         # Freeze CLIP weights
         clip_model.eval()

options/base_option.py

@@ -12,7 +12,7 @@ class BaseOptions():
 
         self.parser.add_argument('--vq_name', type=str, default="rvq_nq1_dc512_nc512", help='Name of the rvq model.')
 
-        self.parser.add_argument("--gpu_id", type=int, default=0, help='GPU id')
+        self.parser.add_argument("--gpu_id", type=int, default=-1, help='GPU id')
         self.parser.add_argument('--dataset_name', type=str, default='t2m', help='Dataset Name, {t2m} for humanml3d, {kit} for kit-ml')
         self.parser.add_argument('--checkpoints_dir', type=str, default='./checkpoints', help='models are saved here.')