Spaces:
Running
on
Zero
Running
on
Zero
update app.py
Browse files
app.py
CHANGED
|
@@ -206,8 +206,6 @@ def downvote(knowledge_base, query):
|
|
| 206 |
|
| 207 |
device = 'cuda'
|
| 208 |
|
| 209 |
-
os.system(f'rm -rf /data-nvme/zerogpu-offload/*')
|
| 210 |
-
|
| 211 |
print("emb model load begin...")
|
| 212 |
model_path = 'openbmb/VisRAG-Ret' # replace with your local model path
|
| 213 |
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|
|
|
|
| 206 |
|
| 207 |
device = 'cuda'
|
| 208 |
|
|
|
|
|
|
|
| 209 |
print("emb model load begin...")
|
| 210 |
model_path = 'openbmb/VisRAG-Ret' # replace with your local model path
|
| 211 |
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|