Merge pull request #1491 from makllama/xd/fix_hf

Fix #1489

Author: rhatdan

ramalama/hf_style_repo_base.py

@@ -182,7 +182,7 @@ def fetch_checksum_from_api(self, organization, file):
         pass
 
     @abstractmethod
-    def create_repository(self, name, organization, tag='latest'):
+    def create_repository(self, name, organization, tag):
         """Create repository instance"""
         pass
 
@@ -328,9 +328,9 @@ def _pull_with_model_store(self, args):
 
         try:
             if not args.quiet:
-                self.print_pull_message(f"{self.get_repo_type()}://{name}:{tag}")
+                self.print_pull_message(f"{self.get_repo_type()}://{organization}/{name}:{tag}")
 
-            repo = self.create_repository(name, organization)
+            repo = self.create_repository(name, organization, tag)
             snapshot_hash = repo.model_hash
             files = repo.get_file_list(cached_files)
             self.store.new_snapshot(tag, snapshot_hash, files)

ramalama/huggingface.py

@@ -165,8 +165,11 @@ def get_repo_type(self):
     def fetch_checksum_from_api(self, organization, file):
         return fetch_checksum_from_api(organization, file)
 
-    def create_repository(self, name, organization, tag='latest'):
-        return HuggingfaceRepositoryModel(name, organization, tag)
+    def create_repository(self, name, organization, tag):
+        if '/' in organization:
+            return HuggingfaceRepositoryModel(name, organization, tag)
+        else:
+            return HuggingfaceRepository(name, organization, tag)
 
     def get_download_url(self, directory, filename):
         return f"https://huggingface.co/{directory}/resolve/main/{filename}"

ramalama/modelscope.py

@@ -80,7 +80,7 @@ def get_repo_type(self):
     def fetch_checksum_from_api(self, organization, file):
         return fetch_checksum_from_api(organization, file)
 
-    def create_repository(self, name, organization, tag='latest'):
+    def create_repository(self, name, organization, tag):
         return ModelScopeRepository(name, organization, tag)
 
     def get_download_url(self, directory, filename):