Update djezzy.py

djezzy.py

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