Revert autocast "fix".

This commit is contained in:
James Betker 2023-04-02 04:04:18 -07:00 committed by GitHub
parent a1d5a27633
commit 25dd04650c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

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.cuda.amp import autocast
from torch import autocast
from tortoise.models.arch_util import normalization, AttentionBlock