Hugging Face's logo

Wendy-Fly
/
Truck2

Model card Files Community
1
Wendy-Fly commited on
Commit
3f48fe9
·
verified ·
1 Parent(s): 440da7b

Upload infer.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. infer.py +1 -1
infer.py CHANGED
@@ -2,7 +2,7 @@ from tqdm import tqdm
2
  #from einops import rearrange
3
  from PIL import Image
4
  from copy import deepcopy
5
- from torchvision.utils import save_image
6
  from typing import List, Optional, Union
7
  from torch import autocast
8
  from torchvision import utils as vutils
 
2
  #from einops import rearrange
3
  from PIL import Image
4
  from copy import deepcopy
5
+
6
  from typing import List, Optional, Union
7
  from torch import autocast
8
  from torchvision import utils as vutils