This commit is contained in:
Daniel Garvey
2023-08-04 14:54:33 -05:00
committed by GitHub
parent a686d7d89f
commit a57eccc997
2 changed files with 8 additions and 2 deletions

View File

@@ -136,7 +136,10 @@ class LoraDataset(Dataset):
if self.center_crop:
crop = min(img.shape[0], img.shape[1])
(h, w,) = (
(
h,
w,
) = (
img.shape[0],
img.shape[1],
)

View File

@@ -144,7 +144,10 @@ def compile_through_fx(
if "cuda" in args.device:
shark_args.enable_tf32 = True
(mlir_module, func_name,) = import_with_fx(
(
mlir_module,
func_name,
) = import_with_fx(
model=model,
inputs=inputs,
is_f16=is_f16,