@@ -1443,7 +1443,7 @@ class HiDreamO1(supported_models_base.BASE):
     }
 
     latent_format = latent_formats.HiDreamO1Pixel
-    memory_usage_factor = 0.6
+    memory_usage_factor = 0.033
     # fp16 not supported: LM MLP down_proj activations fp16 overflow, causing NaNs
     supported_inference_dtypes = [torch.bfloat16, torch.float32]
Original file line number	Diff line number	Diff line change
`@@ -1443,7 +1443,7 @@ class HiDreamO1(supported_models_base.BASE):`
`1443`	`1443`	`}`
`1444`	`1444`
`1445`	`1445`	`latent_format = latent_formats.HiDreamO1Pixel`
`1446`		`- memory_usage_factor = 0.6`
	`1446`	`+ memory_usage_factor = 0.033`
`1447`	`1447`	`# fp16 not supported: LM MLP down_proj activations fp16 overflow, causing NaNs`
`1448`	`1448`	`supported_inference_dtypes = [torch.bfloat16, torch.float32]`
`1449`	`1449`