VincentGOURBIN commited on
Commit
2ac6735
·
verified ·
1 Parent(s): 6cb77ac

Upload step03_chatbot.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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...")