nikigoli commited on
Commit
d92100d
·
verified ·
1 Parent(s): 8e052dd

Turned off use_checkpoint and use_transformer_ckpt for model

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -156,6 +156,8 @@ def build_model_and_transforms(args):
156
 
157
  checkpoint = torch.load(args.pretrain_model_path, map_location="cpu")["model"]
158
  model.load_state_dict(checkpoint, strict=False)
 
 
159
 
160
  model.eval()
161
 
 
156
 
157
  checkpoint = torch.load(args.pretrain_model_path, map_location="cpu")["model"]
158
  model.load_state_dict(checkpoint, strict=False)
159
+ model.use_checkpoint = False
160
+ model.use_transformer_ckpt = False
161
 
162
  model.eval()
163