e58dd86
1
2
3
4
5
6
7
8
#!/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)
