Update djezzy.py

djezzy.py

@@ -28,8 +28,8 @@ embedding_llm = SentenceTransformerEmbeddings(model_name=model_name)
 
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
-tokenizer1 = T5Tokenizer.from_pretrained("google/flan-t5-xl")
-model1 = T5ForConditionalGeneration.from_pretrained("google/flan-t5-xl", device_map=device_map, load_in_8bit=True)
+tokenizer1 = T5Tokenizer.from_pretrained("google/flan-t5-base")
+model1 = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base")
 with tempfile.TemporaryDirectory() as temp_dir:
     # Chemins des fichiers cibles dans le répertoire temporaire
     index_target = os.path.join(temp_dir, 'index.faiss')