mirror of
https://github.com/neonbjb/tortoise-tts.git
synced 2026-01-07 09:10:08 +01:00
Revert autocast "fix".
This commit is contained in:
parent
a1d5a27633
commit
25dd04650c
|
|
@ -5,7 +5,7 @@ from abc import abstractmethod
|
|||
import torch
|
||||
import torch.nn as nn
|
||||
import torch.nn.functional as F
|
||||
from torch.cuda.amp import autocast
|
||||
from torch import autocast
|
||||
|
||||
from tortoise.models.arch_util import normalization, AttentionBlock
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue