neonbjb/tortoise-tts
1
0
Fork 0
mirror of https://github.com/neonbjb/tortoise-tts.git synced 2026-04-21 06:14:04 +00:00
Code Issues Projects Releases Packages 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
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