Merge pull request #380 from serggonin/patch-1

Update diffusion_decoder.py
This commit is contained in:
James Betker 2023-04-01 20:54:45 -06:00 committed by GitHub
commit a1d5a27633
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 import autocast
from torch.cuda.amp import autocast
from tortoise.models.arch_util import normalization, AttentionBlock