Spaces:
Sleeping
Sleeping
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 |
-
|
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(
|