Hugging Face's logo

sunsean
/
Histoformer

Image-to-Image
English
Model card Files Community
sunshangquan commited on
Commit
f6acde2
·
1 Parent(s): 74f47dc

c

Browse files
Files changed (3) hide show
  1. basicsr/models/archs/__pycache__/histoformer_arch.cpython-38.pyc +0 -0
  2. basicsr/models/archs/histoformer_arch.py +1 -1
  3. demo.py +1 -1
basicsr/models/archs/__pycache__/histoformer_arch.cpython-38.pyc CHANGED
Binary files a/basicsr/models/archs/__pycache__/histoformer_arch.cpython-38.pyc and b/basicsr/models/archs/__pycache__/histoformer_arch.cpython-38.pyc differ
 
basicsr/models/archs/histoformer_arch.py CHANGED
@@ -267,7 +267,7 @@ class Upsample(nn.Module):
267
  return self.body(x)
268
 
269
  ##########################################################################
270
- class Histoformer(nn.Module, PyTorchModelHubMixin, repo_url="https://github.com/sunshangquan/Histoformer",):
271
  def __init__(self,
272
  inp_channels=3,
273
  out_channels=3,
 
267
  return self.body(x)
268
 
269
  ##########################################################################
270
+ class Histoformer(nn.Module, PyTorchModelHubMixin, ):
271
  def __init__(self,
272
  inp_channels=3,
273
  out_channels=3,
demo.py CHANGED
@@ -14,4 +14,4 @@ model.load_state_dict(torch.load(filepath, map_location='cpu')['params'], )
14
  model.push_to_hub("sunsean/Histoformer-real")
15
 
16
  # reload
17
- model = ImageCleanModel.from_pretrained("sunsean/Histoformer-real")
 
14
  model.push_to_hub("sunsean/Histoformer-real")
15
 
16
  # reload
17
+ model = Histoformer.from_pretrained("sunsean/Histoformer-real")