From 25dd04650c229bc8f56c353673178a08993a5c9b Mon Sep 17 00:00:00 2001 From: James Betker Date: Sun, 2 Apr 2023 04:04:18 -0700 Subject: [PATCH] Revert autocast "fix". --- tortoise/models/diffusion_decoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tortoise/models/diffusion_decoder.py b/tortoise/models/diffusion_decoder.py index 9d66e87..f67d21a 100644 --- a/tortoise/models/diffusion_decoder.py +++ b/tortoise/models/diffusion_decoder.py @@ -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