mirror of
https://github.com/neonbjb/tortoise-tts.git
synced 2026-04-20 22:04:03 +00:00
Revert autocast "fix".
This commit is contained in:
parent
a1d5a27633
commit
25dd04650c
1 changed files with 1 additions and 1 deletions
|
|
@ -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…
Add table
Add a link
Reference in a new issue