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