Skip to content

Commit a5b78d1

Browse files
committed
Style.
1 parent e97a83e commit a5b78d1

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/lora/utils.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2243,9 +2243,7 @@ def initialize_pipeline(storage_dtype=None, compute_dtype=torch.float32):
22432243
pipe_float8_e4m3_fp32 = initialize_pipeline(storage_dtype=torch.float8_e4m3fn, compute_dtype=torch.float32)
22442244
pipe_float8_e4m3_fp32(**inputs, generator=torch.manual_seed(0))[0]
22452245

2246-
pipe_float8_e4m3_bf16 = initialize_pipeline(
2247-
storage_dtype=torch.float8_e4m3fn, compute_dtype=torch.bfloat16
2248-
)
2246+
pipe_float8_e4m3_bf16 = initialize_pipeline(storage_dtype=torch.float8_e4m3fn, compute_dtype=torch.bfloat16)
22492247
pipe_float8_e4m3_bf16(**inputs, generator=torch.manual_seed(0))[0]
22502248

22512249
@require_peft_version_greater("0.14.0")

0 commit comments

Comments
 (0)