tarekfer8 commited on
Commit
51d1a9e
·
verified ·
1 Parent(s): 9215bda

Update djezzy.py

Browse files
Files changed (1) hide show
  1. djezzy.py +2 -2
djezzy.py CHANGED
@@ -40,8 +40,8 @@ bnb_config =BitsAndBytesConfig(
40
  bnb_8bit_compute_dtype="uint8", # Définissez le type de données de calcul sur un entier non signé de 8 bits
41
  bnb_8bit_use_double_quant=False, # Désactiver la quantification imbriquée (si non applicable)
42
  )
43
- tokenizer1 = T5Tokenizer.from_pretrained("google/flan-t5-xl")
44
- model1 = T5ForConditionalGeneration.from_pretrained("google/flan-t5-xl",quantization_config=bnb_config)
45
  with tempfile.TemporaryDirectory() as temp_dir:
46
  # Chemins des fichiers cibles dans le répertoire temporaire
47
  index_target = os.path.join(temp_dir, 'index.faiss')
 
40
  bnb_8bit_compute_dtype="uint8", # Définissez le type de données de calcul sur un entier non signé de 8 bits
41
  bnb_8bit_use_double_quant=False, # Désactiver la quantification imbriquée (si non applicable)
42
  )
43
+ tokenizer1 = T5Tokenizer.from_pretrained("google/flan-t5-base")
44
+ model1 = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base")
45
  with tempfile.TemporaryDirectory() as temp_dir:
46
  # Chemins des fichiers cibles dans le répertoire temporaire
47
  index_target = os.path.join(temp_dir, 'index.faiss')