Release v0.2.0

Ready to release v0.2.0

Release v0.2.0
Ready to release v0.2.0
ad8097b9 · Muyang Li · GitHub · 804a6d30 · 998192ca · ad8097b9
Unverified Commit ad8097b9 authored Apr 04, 2025 by Muyang Li Committed by GitHub Apr 04, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/utils.py tests/utils.py +1 -1

third_party/Block-Sparse-Attention third_party/Block-Sparse-Attention +1 -1

No files found.
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -59,7 +59,7 @@ class MultiImageDataset(data.Dataset):


 def compute_lpips(
-    ref_dirpath: str, gen_dirpath: str, batch_size: int = 64, num_workers: int = 8, device: str | torch.device = "cuda"
+    ref_dirpath: str, gen_dirpath: str, batch_size: int = 4, num_workers: int = 8, device: str | torch.device = "cuda"
 ) -> float:
    os.environ["TOKENIZERS_PARALLELISM"] = "false"
    metric = LearnedPerceptualImagePatchSimilarity(normalize=True).to(device)

--- a/Block-Sparse-Attention @ 99511c34
+++ b/Block-Sparse-Attention @ 99511c34
-Subproject commit 0d23f715690c5171fd93679de8afd149376db167
+Subproject commit 99511c34554a13ffaa81321834faf66389ffcb30