This commit is contained in:
Johannes Stelzer 2023-02-22 07:50:48 +01:00
parent 3080abc1c4
commit 64d6945c56
1 changed files with 0 additions and 7 deletions

View File

@ -14,9 +14,6 @@
# limitations under the License. # limitations under the License.
import os, sys import os, sys
dp_git = "/home/lugo/git/"
sys.path.append('util')
# sys.path.append('../stablediffusion/ldm')
import torch import torch
torch.backends.cudnn.benchmark = False torch.backends.cudnn.benchmark = False
import numpy as np import numpy as np
@ -25,19 +22,15 @@ warnings.filterwarnings('ignore')
import time import time
import subprocess import subprocess
import warnings import warnings
import torch
from tqdm.auto import tqdm from tqdm.auto import tqdm
from PIL import Image from PIL import Image
# import matplotlib.pyplot as plt # import matplotlib.pyplot as plt
import torch
from movie_util import MovieSaver from movie_util import MovieSaver
import datetime import datetime
from typing import Callable, List, Optional, Union from typing import Callable, List, Optional, Union
import inspect import inspect
from threading import Thread from threading import Thread
torch.set_grad_enabled(False) torch.set_grad_enabled(False)
from omegaconf import OmegaConf
from torch import autocast
from contextlib import nullcontext from contextlib import nullcontext
from ldm.models.diffusion.ddim import DDIMSampler from ldm.models.diffusion.ddim import DDIMSampler