connect with vsc

[Makins18]

from transformers import AutoConfig, AutoModelForSequenceClassification

id2label = {0: "Negative", 1: "Positive"}
label2id = {"Negative": 0, "Positive": 1}

Define checkpoint here to ensure it's in scope

checkpoint = "ArliAI/gpt-oss-20b-Derestricted"

config = AutoConfig.from_pretrained(
checkpoint,
num_labels=2,
id2label=id2label,
label2id=label2id
)

model = AutoModelForSequenceClassification.from_pretrained(
checkpoint,
config=config
)

[Wauplin]

Hi @Makins18 I'm closing this issue as no context is provided. Happy to reopen if you want to explicit your issue

[Makins18]

Reopen