mirror of
https://github.com/SillyTavern/SillyTavern-Extras.git
synced 2026-03-10 14:00:13 +00:00
10 lines
329 B
Python
10 lines
329 B
Python
from typing import Callable, Dict, List
|
|
|
|
from torch import Tensor
|
|
from torch.nn import Module
|
|
|
|
TensorCachedComputationFunc = Callable[
|
|
[Dict[str, Module], List[Tensor], Dict[str, List[Tensor]]], Tensor]
|
|
TensorListCachedComputationFunc = Callable[
|
|
[Dict[str, Module], List[Tensor], Dict[str, List[Tensor]]], List[Tensor]]
|