Update load_meta_data.py

load_meta_data.py

@@ -13,10 +13,8 @@ class ChatBot:
         self.model = None 
     
     def load_from_hub(self,model_id: str):
-        self.tokenizer =  AutoTokenizer.from_pretrained(model_id,
-                                          device_map = 'auto')
-        self.model = AutoModelForCausalLM.from_pretrained(model_id,
-                                             device_map='auto')
+        self.tokenizer =  AutoTokenizer.from_pretrained(model_id,)
+        self.model = AutoModelForCausalLM.from_pretrained(model_id,)
         
     def get_response(self,text: UserQuery) -> str:
         if not self.model or not self.tokenizer: