Spaces:

ai-gero
/

ProtoBind-Diff

Sleeping

manuylo commited on Apr 15

Commit

18c5b6e

verified ·

1 Parent(s): e7732b8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -106,6 +106,7 @@ ckpt_path = hf_hub_download(
 protobind_model = ModelGenerator.load_from_checkpoint(
     ckpt_path,
     map_location=device,
     tokenizer_path=tokenizer_path,
     seq_embedding_dim=1280,
     load=True,

 protobind_model = ModelGenerator.load_from_checkpoint(
     ckpt_path,
     map_location=device,
+    weights_only=False,
     tokenizer_path=tokenizer_path,
     seq_embedding_dim=1280,
     load=True,