dn6 HF staff commited on
Commit
9ebea02
1 Parent(s): 7d224bd
Files changed (1) hide show
  1. README.md +4 -0
README.md CHANGED
@@ -66,6 +66,10 @@ steps), SDXL (50 inference steps), SDXL Turbo (1 inference step) and Würstchen
66
 
67
  ## Code Example
68
 
 
 
 
 
69
  ```shell
70
  pip install diffusers
71
  ```
 
66
 
67
  ## Code Example
68
 
69
+ **Note:** In order to use the `torch.bfloat16` data type with the `StableCascadeDecoderPipeline` you need to have PyTorch 2.2.0 or higher installed. This also means that using the `StableCascadeCombinedPipeline` with `torch.bfloat16` requires PyTorch 2.2.0 or higher, since it calls the StableCascadeDecoderPipeline internally.
70
+
71
+ If it is not possible to install PyTorch 2.2.0 or higher in your environment, the `StableCascadeDecoderPipeline` can be used on its own with the torch.float16 data type. You can download the full precision or bf16 variant weights for the pipeline and cast the weights to torch.float16.
72
+
73
  ```shell
74
  pip install diffusers
75
  ```