Spaces:
Sleeping
Sleeping
Update rag_pipeline.py
Browse files- 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
|