soiz1 commited on
Commit
ee41c57
·
verified ·
1 Parent(s): 1c9751a

Update gpt_sovits/AR/modules/patched_mha_with_cache.py

Browse files
gpt_sovits/AR/modules/patched_mha_with_cache.py CHANGED
@@ -8,7 +8,7 @@ from torch.nn.functional import (
8
  from torch.nn import functional as F
9
  import torch
10
  # Tensor = torch.Tensor
11
- # from typing import Callable, List, Optional, Tuple, Union
12
 
13
 
14
  def multi_head_attention_forward_patched(
 
8
  from torch.nn import functional as F
9
  import torch
10
  # Tensor = torch.Tensor
11
+ from typing import Callable, List, Optional, Tuple, Union
12
 
13
 
14
  def multi_head_attention_forward_patched(