UniPic

Running on Zero

yichenchenchen commited on 17 days ago

Commit

f615fc2

verified ·

1 Parent(s): 1544305

Update inferencer.py

Files changed (1) hide show

inferencer.py CHANGED Viewed

@@ -8,7 +8,7 @@ from torch.nn.utils.rnn import pad_sequence
 import os
 import json
 from mmengine.logging import print_log
 def crop2square(pil_img):
     width, height = pil_img.width, pil_img.height

 import os
 import json
 from mmengine.logging import print_log
+import spaces
 def crop2square(pil_img):
     width, height = pil_img.width, pil_img.height