neonbjb/tortoise-tts
1
0
Fork 0
mirror of https://github.com/neonbjb/tortoise-tts.git synced 2026-01-24 17:40:40 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update diffusion_decoder.py

Browse source 
Chanhged import autocast from torch
This commit is contained in:
Sergey 2023-03-29 13:20:18 +03:00 committed by GitHub
parent c9951ac102
commit 4e990b4327
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tortoise/models/diffusion_decoder.py
View file

@ -5,7 +5,7 @@ from abc import abstractmethod
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch import autocast
from torch.cuda.amp import autocast
from tortoise.models.arch_util import normalization, AttentionBlock