Upload 3 files

config.json

@@ -0,0 +1,279 @@
+{
+  "_id2label": {
+    "intent": [
+      "PlayMusic",
+      "RateBook",
+      "SearchCreativeWork",
+      "SearchScreeningEvent",
+      "AddToPlaylist",
+      "BookRestaurant",
+      "GetWeather"
+    ],
+    "slot": [
+      "O",
+      "B-artist",
+      "I-artist",
+      "B-object_name",
+      "I-object_name",
+      "B-rating_value",
+      "B-rating_unit",
+      "B-sort",
+      "B-music_item",
+      "B-object_type",
+      "I-object_type",
+      "B-best_rating",
+      "B-location_name",
+      "I-location_name",
+      "B-playlist",
+      "I-playlist",
+      "B-movie_name",
+      "I-movie_name",
+      "B-party_size_description",
+      "I-party_size_description",
+      "B-city",
+      "B-restaurant_type",
+      "B-playlist_owner",
+      "B-party_size_number",
+      "I-city",
+      "B-condition_temperature",
+      "B-timeRange",
+      "I-timeRange",
+      "B-country",
+      "B-served_dish",
+      "I-served_dish",
+      "B-movie_type",
+      "B-spatial_relation",
+      "I-spatial_relation",
+      "B-restaurant_name",
+      "I-restaurant_name",
+      "B-entity_name",
+      "I-entity_name",
+      "B-service",
+      "B-cuisine",
+      "B-poi",
+      "I-poi",
+      "I-service",
+      "B-state",
+      "B-track",
+      "I-track",
+      "B-object_select",
+      "B-object_part_of_series_type",
+      "I-restaurant_type",
+      "I-playlist_owner",
+      "B-year",
+      "I-sort",
+      "B-geographic_poi",
+      "I-geographic_poi",
+      "B-genre",
+      "B-object_location_type",
+      "I-object_location_type",
+      "B-current_location",
+      "I-state",
+      "B-condition_description",
+      "I-current_location",
+      "I-country",
+      "I-genre",
+      "I-movie_type",
+      "B-facility",
+      "B-album",
+      "I-cuisine",
+      "I-album",
+      "I-facility",
+      "I-object_part_of_series_type",
+      "I-music_item",
+      "I-object_select"
+    ]
+  },
+  "_label2id": {
+    "intent": {
+      "AddToPlaylist": 4,
+      "BookRestaurant": 5,
+      "GetWeather": 6,
+      "PlayMusic": 0,
+      "RateBook": 1,
+      "SearchCreativeWork": 2,
+      "SearchScreeningEvent": 3
+    },
+    "slot": {
+      "B-album": 65,
+      "B-artist": 1,
+      "B-best_rating": 11,
+      "B-city": 20,
+      "B-condition_description": 59,
+      "B-condition_temperature": 25,
+      "B-country": 28,
+      "B-cuisine": 39,
+      "B-current_location": 57,
+      "B-entity_name": 36,
+      "B-facility": 64,
+      "B-genre": 54,
+      "B-geographic_poi": 52,
+      "B-location_name": 12,
+      "B-movie_name": 16,
+      "B-movie_type": 31,
+      "B-music_item": 8,
+      "B-object_location_type": 55,
+      "B-object_name": 3,
+      "B-object_part_of_series_type": 47,
+      "B-object_select": 46,
+      "B-object_type": 9,
+      "B-party_size_description": 18,
+      "B-party_size_number": 23,
+      "B-playlist": 14,
+      "B-playlist_owner": 22,
+      "B-poi": 40,
+      "B-rating_unit": 6,
+      "B-rating_value": 5,
+      "B-restaurant_name": 34,
+      "B-restaurant_type": 21,
+      "B-served_dish": 29,
+      "B-service": 38,
+      "B-sort": 7,
+      "B-spatial_relation": 32,
+      "B-state": 43,
+      "B-timeRange": 26,
+      "B-track": 44,
+      "B-year": 50,
+      "I-album": 67,
+      "I-artist": 2,
+      "I-city": 24,
+      "I-country": 61,
+      "I-cuisine": 66,
+      "I-current_location": 60,
+      "I-entity_name": 37,
+      "I-facility": 68,
+      "I-genre": 62,
+      "I-geographic_poi": 53,
+      "I-location_name": 13,
+      "I-movie_name": 17,
+      "I-movie_type": 63,
+      "I-music_item": 70,
+      "I-object_location_type": 56,
+      "I-object_name": 4,
+      "I-object_part_of_series_type": 69,
+      "I-object_select": 71,
+      "I-object_type": 10,
+      "I-party_size_description": 19,
+      "I-playlist": 15,
+      "I-playlist_owner": 49,
+      "I-poi": 41,
+      "I-restaurant_name": 35,
+      "I-restaurant_type": 48,
+      "I-served_dish": 30,
+      "I-service": 42,
+      "I-sort": 51,
+      "I-spatial_relation": 33,
+      "I-state": 58,
+      "I-timeRange": 27,
+      "I-track": 45,
+      "O": 0
+    }
+  },
+  "_name_or_path": "OpenSLUv1",
+  "_num_labels": {
+    "intent": 7,
+    "slot": 72
+  },
+  "architectures": [
+    "PretrainedModelForSLUToSave"
+  ],
+  "is_decoder": true,
+  "model": {
+    "_model_target_": "model.OpenSLUModel",
+    "decoder": {
+      "_model_target_": "model.decoder.AGIFDecoder",
+      "intent_classifier": {
+        "_model_target_": "model.decoder.classifier.MLPClassifier",
+        "dropout_rate": 0.4,
+        "ignore_index": -100,
+        "input_dim": 384,
+        "intent_label_num": 7,
+        "loss_fn": {
+          "_model_target_": "torch.nn.BCEWithLogitsLoss"
+        },
+        "mlp": [
+          {
+            "_model_target_": "torch.nn.Linear",
+            "in_features": "{model.encoder.output_dim}",
+            "out_features": 256
+          },
+          {
+            "_model_target_": "torch.nn.LeakyReLU",
+            "negative_slope": 0.2
+          },
+          {
+            "_model_target_": "torch.nn.Linear",
+            "in_features": 256,
+            "out_features": "{base.intent_label_num}"
+          }
+        ],
+        "mode": "intent",
+        "multi_threshold": 0.5,
+        "return_sentence_level": true,
+        "use_intent": true,
+        "use_multi": "{base.multi_intent}",
+        "weight": 0.3
+      },
+      "interaction": {
+        "_model_target_": "model.decoder.interaction.AGIFInteraction",
+        "alpha": 0.2,
+        "dropout_rate": 0.4,
+        "hidden_dim": 128,
+        "input_dim": "{model.encoder.output_dim}",
+        "intent_embedding_dim": 128,
+        "num_heads": 4,
+        "num_layers": 2,
+        "output_dim": "{model.decoder.interaction.intent_embedding_dim}",
+        "row_normalized": true
+      },
+      "slot_classifier": {
+        "_model_target_": "model.decoder.classifier.AutoregressiveLSTMClassifier",
+        "bidirectional": false,
+        "dropout_rate": 0.4,
+        "embedding_dim": 128,
+        "force_ratio": 0.9,
+        "hidden_dim": "{model.decoder.interaction.intent_embedding_dim}",
+        "ignore_index": -100,
+        "input_dim": "{model.encoder.output_dim}",
+        "layer_num": 1,
+        "mode": "slot",
+        "multi_threshold": 0.5,
+        "return_sentence_level": false,
+        "slot_label_num": 72,
+        "use_multi": false,
+        "use_slot": true,
+        "weight": 0.7
+      }
+    },
+    "encoder": {
+      "_model_target_": "model.encoder.AutoEncoder",
+      "attention": {
+        "dropout_rate": 0.4,
+        "hidden_dim": 1024,
+        "output_dim": 128
+      },
+      "embedding": {
+        "dropout_rate": 0.4,
+        "embedding_dim": 128
+      },
+      "encoder_name": "self-attention-lstm",
+      "lstm": {
+        "bidirectional": true,
+        "dropout_rate": 0.4,
+        "layer_num": 1,
+        "output_dim": 256
+      },
+      "output_dim": 384,
+      "return_sentence_level_hidden": true,
+      "return_with_input": true,
+      "unflat_attention": {
+        "dropout_rate": 0.4
+      }
+    }
+  },
+  "return_dict": false,
+  "tokenizer_class": "OpenSLUv1",
+  "torch_dtype": "float32",
+  "transformers_version": "4.25.1",
+  "vocab_size": 11758
+}

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:37952e706657b3dd96f01773b24d3176a6a4406d9d2145234fb284c9597556fa
+size 23961139

tokenizer.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cede86a9c60e470d6e281f1bea2aaadbf8876d573c186fae33da4e007e943b59
+size 205345