add a more meaningful label names

#3
by not-lain - opened
Tunisia.AI org

change LABEL_1 and LABEL_0 to a more meaningful names

labels will be disabled as soon as we enable trust_remote_code
TODO: add custom config_class

Tunisia.AI org

fixed in 577121069d4aab88e160a942e7f709537a281ba6
TODO:

  • fix pipeline
not-lain changed discussion status to closed
not-lain changed discussion status to open
Tunisia.AI org

model.logits contains a list of 4 values, this is due to the original model implementation containing 2 neurons with bias = True
check pic for reference
image

Sign up or log in to comment