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(
|