c

basicsr/models/archs/histoformer_arch.py

@@ -267,7 +267,7 @@ class Upsample(nn.Module):
         return self.body(x)
 
 ##########################################################################
-class Histoformer(nn.Module, PyTorchModelHubMixin):
+class Histoformer(nn.Module, PyTorchModelHubMixin, repo_url="https://github.com/sunshangquan/Histoformer",):
     def __init__(self, 
         inp_channels=3, 
         out_channels=3,