Spaces:
Running
Running
Commit
·
56dc0cd
1
Parent(s):
ff20a1e
Updated.
Browse files
app.py
CHANGED
|
@@ -39,7 +39,6 @@ generation_model = AutoModelForCausalLM.from_pretrained(
|
|
| 39 |
device_map="auto" if torch.cuda.is_available() else None
|
| 40 |
).to(device)
|
| 41 |
|
| 42 |
-
|
| 43 |
# ----------------------
|
| 44 |
# RAG Functions
|
| 45 |
# ----------------------
|
|
|
|
| 39 |
device_map="auto" if torch.cuda.is_available() else None
|
| 40 |
).to(device)
|
| 41 |
|
|
|
|
| 42 |
# ----------------------
|
| 43 |
# RAG Functions
|
| 44 |
# ----------------------
|