Disty0 commited on
Commit
43d0947
·
verified ·
1 Parent(s): 6e745a9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -32,7 +32,7 @@ pipe = diffusers.ZImagePipeline.from_pretrained("Disty0/Z-Image-Turbo-SDNQ-int8"
32
  if triton_is_available and (torch.cuda.is_available() or torch.xpu.is_available()):
33
  pipe.transformer = apply_sdnq_options_to_model(pipe.transformer, use_quantized_matmul=True)
34
  pipe.text_encoder = apply_sdnq_options_to_model(pipe.text_encoder, use_quantized_matmul=True)
35
- pipe.transformer = torch.compile(pipe.transformer)
36
 
37
  pipe.enable_model_cpu_offload()
38
 
 
32
  if triton_is_available and (torch.cuda.is_available() or torch.xpu.is_available()):
33
  pipe.transformer = apply_sdnq_options_to_model(pipe.transformer, use_quantized_matmul=True)
34
  pipe.text_encoder = apply_sdnq_options_to_model(pipe.text_encoder, use_quantized_matmul=True)
35
+ pipe.transformer = torch.compile(pipe.transformer) # optional for faster speeds
36
 
37
  pipe.enable_model_cpu_offload()
38