Spaces:
Running
on
Zero
Running
on
Zero
Upload step03_chatbot.py with huggingface_hub
Browse files- step03_chatbot.py +4 -0
step03_chatbot.py
CHANGED
|
@@ -510,6 +510,10 @@ class GenericRAGChatbot:
|
|
| 510 |
print(" 📥 Chargement des embeddings SafeTensors...")
|
| 511 |
tensors = load_file(embeddings_file)
|
| 512 |
embeddings_tensor = tensors["embeddings"]
|
|
|
|
|
|
|
|
|
|
|
|
|
| 513 |
embeddings_np = embeddings_tensor.numpy().astype(np.float32)
|
| 514 |
|
| 515 |
print(" 📋 Chargement des métadonnées...")
|
|
|
|
| 510 |
print(" 📥 Chargement des embeddings SafeTensors...")
|
| 511 |
tensors = load_file(embeddings_file)
|
| 512 |
embeddings_tensor = tensors["embeddings"]
|
| 513 |
+
|
| 514 |
+
# Assurer que le tensor est sur CPU pour la conversion numpy
|
| 515 |
+
if embeddings_tensor.is_cuda:
|
| 516 |
+
embeddings_tensor = embeddings_tensor.cpu()
|
| 517 |
embeddings_np = embeddings_tensor.numpy().astype(np.float32)
|
| 518 |
|
| 519 |
print(" 📋 Chargement des métadonnées...")
|