bfloat16

sayakpaul · sayakpaul · commit 8ca260dc1b86 · 2025-01-10T20:12:14.000+05:30
diff --git a/docs/source/en/api/pipelines/hunyuan_video.md b/docs/source/en/api/pipelines/hunyuan_video.md
@@ -48,7 +48,7 @@ transformer_8bit = HunyuanVideoTransformer3DModel.from_pretrained(
     "hunyuanvideo-community/HunyuanVideo",
     subfolder="transformer",
     quantization_config=quant_config,
-    torch_dtype=torch.float16,
+    torch_dtype=torch.bfloat16,
 )
 
 pipeline = HunyuanVideoPipeline.from_pretrained(

Original file line number	Diff line number	Diff line change
`@@ -48,7 +48,7 @@ transformer_8bit = HunyuanVideoTransformer3DModel.from_pretrained(`
`48`	`48`	`"hunyuanvideo-community/HunyuanVideo",`
`49`	`49`	`subfolder="transformer",`
`50`	`50`	`quantization_config=quant_config,`
`51`		`- torch_dtype=torch.float16,`
	`51`	`+ torch_dtype=torch.bfloat16,`
`52`	`52`	`)`
`53`	`53`
`54`	`54`	`pipeline = HunyuanVideoPipeline.from_pretrained(`