mb-rag 1.1.23__tar.gz → 1.1.24__tar.gz

{mb_rag-1.1.23 → mb_rag-1.1.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: mb_rag
-Version: 1.1.23
+Version: 1.1.24
 Summary: RAG function file
 Author: ['Malav Bateriwala']
 Requires-Python: >=3.8

{mb_rag-1.1.23 → mb_rag-1.1.24}/mb_rag/chatbot/basic.py

@@ -218,12 +218,12 @@ class ModelFactory:
         if not check_package("transformers"):
             raise ImportError("Transformers package not found. Please install it using: pip install transformers")
         if not check_package("langchain_huggingface"):
-            raise ImportError("Transformers package not found. Please install it using: pip install langchain_huggingface")
+            raise ImportError("langchain_huggingface package not found. Please install it using: pip install langchain_huggingface")
         if not check_package("torch"):
             raise ImportError("Torch package not found. Please install it using: pip install torch")
 
         from langchain_huggingface import HuggingFacePipeline
-        from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, AutoModelForImageTextToText
+        from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, AutoModelForImageTextToText,AutoProcessor
         import torch
         
         device = torch.device(device) if torch.cuda.is_available() else torch.device("cpu")
@@ -231,8 +231,8 @@ class ModelFactory:
         temperature = kwargs.pop("temperature", 0.7)
         max_length = kwargs.pop("max_length", 1024)
         
-        tokenizer = AutoTokenizer.from_pretrained(model_name,trust_remote_code=True)
         if model_function == "image-text-to-text":
+            tokenizer = AutoProcessor.from_pretrained(model_name,trust_remote_code=True)
             model = AutoModelForImageTextToText.from_pretrained(
                 model_name,
                 torch_dtype=torch.float16 if device == "cuda" else torch.float32,
@@ -241,6 +241,7 @@ class ModelFactory:
                 **kwargs
             )
         else:
+            tokenizer = AutoTokenizer.from_pretrained(model_name,trust_remote_code=True)
             model = AutoModelForCausalLM.from_pretrained(
                 model_name,
                 torch_dtype=torch.float16 if device == "cuda" else torch.float32,

{mb_rag-1.1.23 → mb_rag-1.1.24}/mb_rag/version.py

@@ -1,5 +1,5 @@
 MAJOR_VERSION = 1
 MINOR_VERSION = 1
-PATCH_VERSION = 23
+PATCH_VERSION = 24
 version = '{}.{}.{}'.format(MAJOR_VERSION, MINOR_VERSION, PATCH_VERSION)
 __all__  = ['MAJOR_VERSION', 'MINOR_VERSION', 'PATCH_VERSION', 'version']

{mb_rag-1.1.23 → mb_rag-1.1.24}/mb_rag.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: mb_rag
-Version: 1.1.23
+Version: 1.1.24
 Summary: RAG function file
 Author: ['Malav Bateriwala']
 Requires-Python: >=3.8