mirror of
https://github.com/SillyTavern/SillyTavern-Extras.git
synced 2026-04-26 09:28:57 +00:00
10 lines
252 B
Python
10 lines
252 B
Python
from torch.nn import Module
|
|
from torch.nn.utils import spectral_norm
|
|
|
|
|
|
def apply_spectral_norm(module: Module, use_spectrial_norm: bool = False) -> Module:
|
|
if use_spectrial_norm:
|
|
return spectral_norm(module)
|
|
else:
|
|
return module
|