[Bugfix] Fix dynamic FP8 quantization for Mixtral (#4793)

This commit is contained in:
Philipp Moritz authored and GitHub committed 2024-05-13 19:00:27 -04:00
1 parent 1356df53bd
commit 33d3914b1e
1 file changed
+1 -1
+1 -1
View File
@@ -95,7 +95,7 @@ class MixtralMoE(nn.Module):
params_dtype=self.params_dtype,
quant_config=None)
if self.use_fp8:
if self.use_fp8 and self.quant_config.is_checkpoint_fp8_serialized:
params_dtype = torch.float8_e4m3fn
self.w13_weight = nn.Parameter(