feat(modeling_stablelm_epoch.py): add support for AutoModelForSequenceClassification

by caddiehealth - opened Jan 23

base: refs/heads/main

←

from: refs/pr/3

Discussion Files changed

+113

-4

feat(modeling_stablelm_epoch.py): add support for AutoModelForSequenceClassificationdbab9764

caddiehealth

Jan 23

•

edited Jan 23

Adds support for using AutoModelForSequenceClassification. Currently you would receive an error like this: Unrecognized configuration class for this kind of AutoModel: AutoModelForSequenceClassification.

There aren't tests setup here (perhaps we could add) but here's an example showing that it starts training as-expected

add support in config.json8165cdbb

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Cannot merge

This branch has merge conflicts in the following files:

config.json
modeling_stablelm_epoch.py

· Sign up or log in to comment