sayande commited on
Commit
5e65a8e
·
verified ·
1 Parent(s): 8a0e370

Update rag_pipeline.py

Browse files
Files changed (1) hide show
  1. rag_pipeline.py +1 -0
rag_pipeline.py CHANGED
@@ -9,6 +9,7 @@ import json
9
  import sqlite3
10
  import uuid
11
  from datetime import datetime
 
12
  import torch
13
  from transformers import AutoTokenizer, AutoModelForCausalLM
14
  from peft import PeftModel
 
9
  import sqlite3
10
  import uuid
11
  from datetime import datetime
12
+ from typing import List, Dict, Any
13
  import torch
14
  from transformers import AutoTokenizer, AutoModelForCausalLM
15
  from peft import PeftModel