Spaces:

jykoh
/

fromage

Build error

jykoh commited on Feb 20, 2023

Commit

0b88c5c

1 Parent(s): a03fe94

Fix typo

Files changed (1) hide show

fromage/models.py CHANGED Viewed

@@ -644,7 +644,7 @@ def load_fromage(embeddings_dir: str, model_args_path: str, model_ckpt_path: str
   model = model.bfloat16()
   model = model.cuda()
-  Load pretrained linear mappings and [RET] embeddings.
   checkpoint = torch.load(model_ckpt_path)
   model.load_state_dict(checkpoint['state_dict'], strict=False)
   with torch.no_grad():

   model = model.bfloat16()
   model = model.cuda()
+  # Load pretrained linear mappings and [RET] embeddings.
   checkpoint = torch.load(model_ckpt_path)
   model.load_state_dict(checkpoint['state_dict'], strict=False)
   with torch.no_grad():