#!/usr/bin/env python3
from deepfloyd_if.modules import IFStageI
from time import time

start_time = time()
model = IFStageI(dir_or_name="/home/patrick/IF-I-IF-v1.0/", device="cuda")
print("Time", time() - start_time)