Update nlu/models/slot_filler.py

This commit is contained in:
Med Marrouchi 2024-11-28 19:47:17 +01:00 committed by GitHub
parent d7f69ff2be
commit 5a522b14cc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -202,10 +202,6 @@ class SlotFiller(tfbp.Model):
token = tokens[idx] token = tokens[idx]
slot_id = slot_ids[idx] slot_id = slot_ids[idx]
# Skip special tokens
if token in special_tokens:
idx += 1
continue
# Get slot name # Get slot name
slot_name = self.extra_params["slot_names"][slot_id] slot_name = self.extra_params["slot_names"][slot_id]