alexnasa commited on
Commit
298370d
·
verified ·
1 Parent(s): bb076a6

Update pipelines/pipeline_seesr.py

Browse files
Files changed (1) hide show
  1. pipelines/pipeline_seesr.py +1 -0
pipelines/pipeline_seesr.py CHANGED
@@ -22,6 +22,7 @@ import numpy as np
22
  import PIL.Image
23
  import torch
24
  import torch.nn.functional as F
 
25
  from torchvision.utils import save_image
26
  from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer
27
 
 
22
  import PIL.Image
23
  import torch
24
  import torch.nn.functional as F
25
+ from torch.nn.functional import unfold, fold
26
  from torchvision.utils import save_image
27
  from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer
28