set model.transformer not model. for checkpointing

app.py

@@ -156,8 +156,8 @@ def build_model_and_transforms(args):
 
     checkpoint = torch.load(args.pretrain_model_path, map_location="cpu")["model"]
     model.load_state_dict(checkpoint, strict=False)
-    model.use_checkpoint = False
-    model.use_transformer_ckpt = False
+    model.transformer.use_checkpoint = False
+    model.transformer.use_transformer_ckpt = False
 
     model.eval()