Update modeling_captionbert.py

modeling_captionbert.py

@@ -315,9 +315,16 @@ def _attach_amoe(self, arms=None, dispatch=None, repo=None, tau=None, verify=Tru
             present = set()
         dispatch = next((c for c in AMOE_DISPATCH_CANDIDATES if c in present), None)
         if dispatch is None:
+            local = os.path.isdir(str(repo))
+            hint = (f"'{repo}' is a LOCAL DIRECTORY, so there is no hub repo to "
+                    f"search. Pass the repo explicitly:\n"
+                    f"    model.attach_amoe(repo='AbstractPhil/captionbert-8192-v2-B')"
+                    if local else
+                    f"Pass dispatch=<path in {repo}>, or dispatch=False to attach "
+                    f"an UNTRAINED dispatch.")
             raise FileNotFoundError(
-                f"no dispatch found in {repo}. Tried {list(AMOE_DISPATCH_CANDIDATES)}. "
-                f"Pass dispatch=<path>, or dispatch=False for an untrained one.")
+                f"no dispatch found in {repo}. Tried "
+                f"{list(AMOE_DISPATCH_CANDIDATES)}.\n         {hint}")
         print(f"[amoe] dispatch: {repo}/{dispatch}")
     if dispatch is False:
         dispatch = None