tarekfer8 commited on
Commit
5b6a015
·
verified ·
1 Parent(s): 2f2cdfe

Update djezzy.py

Browse files
Files changed (1) hide show
  1. djezzy.py +3 -0
djezzy.py CHANGED
@@ -29,6 +29,9 @@ model_name = "sentence-transformers/all-MiniLM-L6-v2"
29
  embedding_llm = SentenceTransformerEmbeddings(model_name=model_name)
30
 
31
  from transformers import T5Tokenizer, T5ForConditionalGeneration
 
 
 
32
  compute_dtype = getattr(torch, bnb_8bit_compute_dtype)
33
 
34
  bnb_config =BitsAndBytesConfig(
 
29
  embedding_llm = SentenceTransformerEmbeddings(model_name=model_name)
30
 
31
  from transformers import T5Tokenizer, T5ForConditionalGeneration
32
+ bnb_8bit_compute_dtype="uint8"
33
+ # Activer la quantification imbriquée pour les modèles de base 8 bits
34
+ use_nested_quant = True
35
  compute_dtype = getattr(torch, bnb_8bit_compute_dtype)
36
 
37
  bnb_config =BitsAndBytesConfig(