chore: Update package names and dependencies

endomorphosis · Sep 2, 2024 · d576d20 · d576d20
1 parent ec50ee5
commit d576d20
Show file tree

Hide file tree

Showing 9 changed files with 5 additions and 4 deletions.
diff --git a/example.py b/example.py
@@ -1,5 +1,5 @@
 from transformers import AutoModel
-from ipfs_transformers import AutoModel
+from ipfs_transformers_py import AutoModel
 
 model = AutoModel.from_auto_download("bge-small-en-v1.5")  
 print(dir(model))

diff --git a/ipfs_transformers/__init__.py → ipfs_transformers_py/__init__.py b/ipfs_transformers/__init__.py → ipfs_transformers_py/__init__.py
diff --git a/ipfs_transformers/ipfs_kit.py → ipfs_transformers_py/ipfs_kit.py b/ipfs_transformers/ipfs_kit.py → ipfs_transformers_py/ipfs_kit.py
diff --git a/ipfs_transformers/ipfs_transformers.py → ipfs_transformers_py/ipfs_transformers.py b/ipfs_transformers/ipfs_transformers.py → ipfs_transformers_py/ipfs_transformers.py
diff --git a/...ansformers/ipfs_transformers_generator.py → ...formers_py/ipfs_transformers_generator.py b/...ansformers/ipfs_transformers_generator.py → ...formers_py/ipfs_transformers_generator.py
diff --git a/...ransformers/ipfs_transformers_template.py → ...sformers_py/ipfs_transformers_template.py b/...ransformers/ipfs_transformers_template.py → ...sformers_py/ipfs_transformers_template.py
@@ -60,7 +60,7 @@ def __init__(self, collection=None, meta=None):
 					"s3_cfg": self.s3cfg,
 					"role": self.role
 				}
-		from ipfs_transformers.model_manager_bak import model_manager as model_manager
+		from ipfs_transformers_py.model_manager_bak import model_manager as model_manager
 		self.model_manager = model_manager(collection, meta)
 		self.model_manager.load_collection_cache()
 		self.model_manager.state()

diff --git a/ipfs_transformers/test/test_fio.py → ipfs_transformers_py/test/test_fio.py b/ipfs_transformers/test/test_fio.py → ipfs_transformers_py/test/test_fio.py
diff --git a/ipfs_transformers/test/test_hf_ipfs.py → ipfs_transformers_py/test/test_hf_ipfs.py b/ipfs_transformers/test/test_hf_ipfs.py → ipfs_transformers_py/test/test_hf_ipfs.py
@@ -16,7 +16,7 @@ def test(self, **kwargs):
         from transformers import AutoModel
         #model = AutoModel.from_pretrained("google-bert/bert-base-cased")
         #what do we want to call the library?
-        from ipfs_transformers import AutoModel
+        from ipfs_transformers_py import AutoModel
         #do we want to use the ipfs hash or the model name? This requires IPNS to be enabled
         #model = T5Model.from_ipfs("QmWJr4M1VN5KpJjqCsJsJg7PDmFoqQYs1BKpYxcdMY1qkh")
         #model = AutoModelForSeq2SeqLM.from_auto_download("t0") # 40GB test this afterwards

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,5 @@
-ipfs_kit@git+https://github.com/endomorphosis/ipfs_kit.git
+ipfs_kit_py
+ipfs_model_manager_py
 transformers
 torch
 torchvision