We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e97a83e commit a5b78d1Copy full SHA for a5b78d1
tests/lora/utils.py
@@ -2243,9 +2243,7 @@ def initialize_pipeline(storage_dtype=None, compute_dtype=torch.float32):
2243
pipe_float8_e4m3_fp32 = initialize_pipeline(storage_dtype=torch.float8_e4m3fn, compute_dtype=torch.float32)
2244
pipe_float8_e4m3_fp32(**inputs, generator=torch.manual_seed(0))[0]
2245
2246
- pipe_float8_e4m3_bf16 = initialize_pipeline(
2247
- storage_dtype=torch.float8_e4m3fn, compute_dtype=torch.bfloat16
2248
- )
+ pipe_float8_e4m3_bf16 = initialize_pipeline(storage_dtype=torch.float8_e4m3fn, compute_dtype=torch.bfloat16)
2249
pipe_float8_e4m3_bf16(**inputs, generator=torch.manual_seed(0))[0]
2250
2251
@require_peft_version_greater("0.14.0")
0 commit comments