Upload 2 files

NonReference_RGBN_x4/load.py

@@ -29,7 +29,6 @@ config = {
         "attn_resolutions": [],
         "dropout": 0.0
     },
-
     "cond_stage_config": {
         "image_size": 64,
         "in_channels": 8,
@@ -40,7 +39,6 @@ config = {
         "channel_mult": [1, 2, 2, 4],
         "num_head_channels": 32
     },
-
     "other": {
         "concat_mode": True,
         "cond_stage_trainable": False,
@@ -56,7 +54,7 @@ def example_data(path: pathlib.Path, device: str = "cuda:0", *args, **kwargs):
     sample = safetensors.torch.load_file(data_f)
     return  sample["lr"].to(device), sample["hr"].to(device)
 
-def trainable_model(path, device: str = "cuda:0", *args, **kwargs):
+def trainable_model(*args, **kwargs):
     print("Trainable model is not available for this model.")
     return None
 

NonReference_RGBN_x4/mlm.json

@@ -119,7 +119,7 @@
     "links": [],
     "assets": {
         "trainable": {
-            "href": "https://huggingface.co/tacofoundation/RS-SR-LTDF/resolve/main/NonReference_RGBN_x4/model.safetensor",
+            "href": "https://huggingface.co/tacofoundation/tacofoundation/resolve/main/RS-SR-LTDF/NonReference_RGBN_x4/model.safetensor",
             "type": "application/octet-stream; application=safetensor",
             "title": "Pytorch model weights checkpoint",
             "description": "The weights of the model in safetensor format.",
@@ -131,7 +131,7 @@
             ]
         },
         "hardconstraint": {
-            "href": "https://huggingface.co/tacofoundation/RS-SR-LTDF/resolve/main/NonReference_RGBN_x4/hard_constraint.safetensor",
+            "href": "https://huggingface.co/tacofoundation/tacofoundation/resolve/main/RS-SR-LTDF/NonReference_RGBN_x4/hard_constraint.safetensor",
             "type": "application/octet-stream; application=safetensor",
             "title": "Pytorch hard constraint weights checkpoint. It is used to load faster the hard constraint module.",
             "description": "The weights of the model in safetensor format.",
@@ -143,7 +143,7 @@
             ]
         },
         "source_code": {
-            "href": "https://huggingface.co/tacofoundation/RS-SR-LTDF/resolve/main/NonReference_RGBN_x4/load.py",
+            "href": "https://huggingface.co/tacofoundation/tacofoundation/resolve/main/RS-SR-LTDF/NonReference_RGBN_x4/load.py",
             "type": "text/x-python",
             "title": "Model load script",
             "description": "Python script to load the model.",
@@ -153,7 +153,7 @@
             ]
         },
         "example_data": {
-            "href": "https://huggingface.co/tacofoundation/RS-SR-LTDF/resolve/main/NonReference_RGBN_x4/example_data.safetensor",
+            "href": "https://huggingface.co/tacofoundation/tacofoundation/resolve/main/RS-SR-LTDF/NonReference_RGBN_x4/example_data.safetensor",
             "type": "application/octet-stream; application=safetensors",
             "title": "Example Sentinel-2 image",
             "description": "Example Sentinel-2 image for model inference.",