Fix the mistake of not importing the gc (#1939)

This commit is contained in:
rmagur1203
2022-12-12 23:14:09 +09:00
committed by GitHub
parent e159bb3dce
commit 96a12099ed

View File

@ -5,6 +5,7 @@ ldm.invoke.generator.txt2img inherits from ldm.invoke.generator
import torch import torch
import numpy as np import numpy as np
import math import math
import gc
from ldm.invoke.generator.base import Generator from ldm.invoke.generator.base import Generator
from ldm.models.diffusion.ddim import DDIMSampler from ldm.models.diffusion.ddim import DDIMSampler
from ldm.invoke.generator.omnibus import Omnibus from ldm.invoke.generator.omnibus import Omnibus