OzzyGT HF Staff commited on
Commit
f431ea7
·
verified ·
1 Parent(s): 320b120

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -58,11 +58,14 @@ You can find ready-to-use scripts in the [diffusers-recipes](https://github.com/
58
 
59
  ```python
60
  import torch
61
- from diffusers import ModularPipeline
62
  from diffusers.hooks import apply_group_offloading
63
  from diffusers.utils.export_utils import encode_video
64
  from sdnq import SDNQConfig # noqa: F401
65
 
 
 
 
66
  pipe = ModularPipeline.from_pretrained("OzzyGT/MiniMax_H3_sdnq_4bit_pruned", workflow="t2va", trust_remote_code=True)
67
  pipe.load_components(dtype=torch.bfloat16, trust_remote_code=True)
68
 
 
58
 
59
  ```python
60
  import torch
61
+ from diffusers import AutoencoderKLMiniMaxH3, ModularPipeline
62
  from diffusers.hooks import apply_group_offloading
63
  from diffusers.utils.export_utils import encode_video
64
  from sdnq import SDNQConfig # noqa: F401
65
 
66
+ # patch until fixed upstream, the fp32 pin covers the whole VAE and ignores dtype
67
+ AutoencoderKLMiniMaxH3._keep_in_fp32_modules = []
68
+
69
  pipe = ModularPipeline.from_pretrained("OzzyGT/MiniMax_H3_sdnq_4bit_pruned", workflow="t2va", trust_remote_code=True)
70
  pipe.load_components(dtype=torch.bfloat16, trust_remote_code=True)
71