Update README.md

README.md

@@ -7,21 +7,5 @@ base_model_relation: quantized
 
 FP8 quantized version of [AuraFlow v0.3](fal/AuraFlow-v0.3)
 
-## Quantization
+Just casted to `torch.float8_e4m3fn` all linear weights of the flow transformer except `t_embedder`, `final_linear`, `modF`.
 
-```py
-import torch
-from huggingface_hub import cached_download
-from safetensors.torch import load_file, save_file
-
-ckpt_path = cached_download(
-    "https://huggingface.co/fal/AuraFlow-v0.3/resolve/main/aura_flow_0.3.safetensors",
-)
-
-state_dict = load_file(ckpt_path)
-
-for key, value in state_dict.items():
-    state_dict[key] = value.to(torch.float8_e4m3fn)
-
-save_file(state_dict, "./aura_flow_0.3.float8_e4m3fn.safetensors")
-```