Skip to content

GitLab

Unverified Commit ad8097b9 authored by Muyang Li's avatar Muyang Li Committed by GitHub
Browse files

Release v0.2.0 

Ready to release v0.2.0
parents 804a6d30 998192ca
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/utils.py
View file @ ad8097b9
...@@ -59,7 +59,7 @@ class MultiImageDataset(data.Dataset): ...@@ -59,7 +59,7 @@ class MultiImageDataset(data.Dataset):
def compute_lpips( 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: ) -> float:
os.environ["TOKENIZERS_PARALLELISM"] = "false" os.environ["TOKENIZERS_PARALLELISM"] = "false"
metric = LearnedPerceptualImagePatchSimilarity(normalize=True).to(device) metric = LearnedPerceptualImagePatchSimilarity(normalize=True).to(device)
......
Block-Sparse-Attention @ 99511c34
Compare 0d23f715...99511c34
Subproject commit 0d23f715690c5171fd93679de8afd149376db167 Subproject commit 99511c34554a13ffaa81321834faf66389ffcb30
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment