c

basicsr/models/archs/histoformer_arch.py

@@ -271,11 +271,11 @@ class Histoformer(nn.Module):
     def __init__(self, 
         inp_channels=3, 
         out_channels=3, 
-        dim = 48,
-        num_blocks = [4,6,6,8], 
+        dim = 36,
+        num_blocks = [4,4,6,8], 
         num_refinement_blocks = 4,
         heads = [1,2,4,8],
-        ffn_expansion_factor = 2.66,
+        ffn_expansion_factor = 2.667,
         bias = False,
         LayerNorm_type = 'WithBias',   ## Other option 'BiasFree'
         dual_pixel_task = False        ## True for dual-pixel defocus deblurring only. Also set inp_channels=6

demo.py

@@ -1,18 +1,15 @@
-from basicsr.model.image_restoration_model import ImageCleanModel
-<<<<<<< HEAD
-from huggingface_hub import PyTorchModelHubMixin
-=======
->>>>>>> 0c10de556292aefe29d7af74311340a7eb844834
+from basicsr.models.archs.histoformer_arch import Histoformer
+# from huggingface_hub import PyTorchModelHubMixin
 
 # define model
-model = ImageCleanModel(...)
+model = Histoformer()
 
 # equip with weights
-filepath = hf_hub_download(repo_id="sunsean/Histoformer", filename="net_g_best.pth")
-model.load_state_dict(filepath, map_location="cpu")
+filepath = hf_hub_download(repo_id="sunsean/Histoformer", filename="Allweather/pretrained_models/net_g_real.pth")
+model.load_state_dict(filepath, map_location="gpu")
 
 # push to hub
-model.push_to_hub("sunsean/Histoformer-best")
+model.push_to_hub("sunsean/Histoformer-real")
 
 # reload
-model = ImageCleanModel.from_pretrained("sunsean/Histoformer-best")
+model = ImageCleanModel.from_pretrained("sunsean/Histoformer-real")

git.bash

@@ -1,3 +1,3 @@
 git add .
-git commit -m "commit from $USER"
+git commit -m "c"
 git push