Treat hf.co/ prefix the same as hf://

ollama uses hf.co/ to specify huggingface prefix, like RamaLama
uses hf://

Treat them similarly.

Signed-off-by: Eric Curtin <[email protected]>

Author: ericcurtin

examples/run/run.cpp

@@ -640,7 +640,17 @@ class LlamaData {
             return 1;
         }
 
-        model_ = model_.substr(pos + 3);  // Skip past "://"
+        model_ = model_.substr(pos + sizeof("://") - 1);  // Skip past "://"
+        return 0;
+    }
+
+    int remove_hf_co(std::string & model_) {
+        const std::string::size_type pos = model_.find("hf.co/");
+        if (pos == std::string::npos) {
+            return 1;
+        }
+
+        model_ = model_.substr(pos + sizeof("hf.co/") - 1);  // Skip past "hf.co/"
         return 0;
     }
 
@@ -658,11 +668,14 @@ class LlamaData {
         if (string_starts_with(model_, "hf://") || string_starts_with(model_, "huggingface://")) {
             remove_proto(model_);
             ret = huggingface_dl(model_, headers, bn);
+        } else if (string_starts_with(model_, "hf.co/")) {
+            remove_hf_co(model_);
+            ret = huggingface_dl(model_, headers, bn);
         } else if (string_starts_with(model_, "ollama://")) {
             remove_proto(model_);
             ret = ollama_dl(model_, headers, bn);
         } else if (string_starts_with(model_, "https://")) {
-            download(model_, headers, bn, true);
+            ret = download(model_, headers, bn, true);
         } else {
             ret = ollama_dl(model_, headers, bn);
         }