Update ai_assistant.py

ai_assistant.py

@@ -44,8 +44,8 @@ def build_qa():
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForCausalLM.from_pretrained(
             model_name,
-            device_map="auto",
-            torch_dtype=torch.float16,
+            device_map="auto",   # works after installing accelerate
+            dtype=torch.float16, # instead of torch_dtype
         )
         llm = TextGenerationPipeline(
             model=model,