This commit is contained in:
Johannes Stelzer 2024-01-09 15:58:56 +01:00
parent 70aa6f0a76
commit 7d3fa4b3a0
1 changed files with 1 additions and 2 deletions

View File

@ -17,9 +17,8 @@ import torch
import warnings import warnings
from latent_blending import LatentBlending from latent_blending import LatentBlending
from diffusers_holder import DiffusersHolder from diffusers_holder import DiffusersHolder
from diffusers import DiffusionPipeline
from diffusers import AutoPipelineForText2Image from diffusers import AutoPipelineForText2Image
warnings.filterwarnings('ignore') warnings.filterwarnings('ignore')
torch.set_grad_enabled(False) torch.set_grad_enabled(False)
torch.backends.cudnn.benchmark = False torch.backends.cudnn.benchmark = False