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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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