File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed 
docs/source/en/api/pipelines Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -59,10 +59,10 @@ Refer to the [Quantization](../../quantization/overview) overview to learn more
5959``` py 
6060import  torch
6161from  diffusers import  BitsAndBytesConfig as  DiffusersBitsAndBytesConfig, SanaTransformer2DModel, SanaPipeline
62- from  transformers import  BitsAndBytesConfig as  BitsAndBytesConfig, AutoModelForCausalLM 
62+ from  transformers import  BitsAndBytesConfig as  BitsAndBytesConfig, AutoModel 
6363
6464quant_config =  BitsAndBytesConfig(load_in_8bit = True )
65- text_encoder_8bit =  AutoModelForCausalLM .from_pretrained(
65+ text_encoder_8bit =  AutoModel .from_pretrained(
6666    " Efficient-Large-Model/Sana_1600M_1024px_diffusers" 
6767    subfolder = " text_encoder" 
6868    quantization_config = quant_config,
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments