#!/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) | |
#!/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) | |