diffusers/tools
1127
1#!/usr/bin/env python32from deepfloyd_if.modules import IFStageI3from time import time4 5start_time = time()6model = IFStageI(dir_or_name="/home/patrick/IF-I-IF-v1.0/", device="cuda")7print("Time", time() - start_time)8 1#!/usr/bin/env python32from deepfloyd_if.modules import IFStageI3from time import time4 5start_time = time()6model = IFStageI(dir_or_name="/home/patrick/IF-I-IF-v1.0/", device="cuda")7print("Time", time() - start_time)8