From 4f59e4ac200b1d7a2fab94493ee141b5d776d456 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Sun, 5 Jul 2020 12:43:42 -0700 Subject: [PATCH] update readme --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 0f30eae..54f8036 100644 --- a/README.md +++ b/README.md @@ -16,7 +16,7 @@ $ pip install local-attention ```python import torch -from local_attention.local_attention import LocalAttention +from local_attention import LocalAttention q = torch.randn(8, 2048, 64) k = torch.randn(8, 2048, 64) @@ -38,7 +38,7 @@ This library also allows for local attention in the setting of shared query/key ```python import torch -from local_attention.local_attention import LocalAttention +from local_attention import LocalAttention qk = torch.randn(8, 2048, 64) v = torch.randn(8, 2048, 64)