Add mini model for testing

config_jsons/mini_model.json

@@ -0,0 +1,31 @@
+{
+    "angles_definitions": "canonical-full-angles",
+    "zero_center": true,
+    "timesteps": 250,
+    "max_seq_len": 128,
+    "min_seq_len": 40,
+    "trim_strategy": "discard",
+    "variance_schedule": "cosine",
+    "variance_scale": 1.0,
+    "time_encoding": "gaussian_fourier",
+    "position_embedding_type": "relative_key",
+    "num_hidden_layers": 6,
+    "hidden_size": 192,
+    "intermediate_size": 384,
+    "num_heads": 6,
+    "dropout_p": 0.1,
+    "decoder": "mlp",
+    "gradient_clip": 1.0,
+    "lr": 5e-5,
+    "loss": "smooth_l1",
+    "l2_norm": 0.0,
+    "l1_norm": 0.0,
+    "circle_reg": 0.0,
+    "lr_scheduler": "LinearWarmup",
+    "min_epochs": 20,
+    "max_epochs": 20,
+    "early_stop_patience": 0,
+    "use_swa": false,
+    "batch_size": 128,
+    "multithread": true
+}

setup_esm_inverse_folding_env.sh

@@ -1,3 +0,0 @@
-mamba create -n inverse -c conda-forge pytorch_geometric pytorch-gpu=1.10
-pip install biotite
-pip install git+https://github.com/facebookresearch/esm.git