jannesscholz issues

Repositories
Issues
Comments

Results 1 issues of


                                            jannesscholz

[FR] Provide load arguments to model.from_pretrained when calling mlflow.transformers.load_model

### Willingness to contribute Yes. I can contribute this feature independently. ### Proposal Summary I finetuned a Llama 3.1 8B on a 6-class-classification task (loaded with AutoModelForSequenceClassification.from_pretrained("meta-llama/Meta-Llama-3.1-8B", num_labels = 6)...

enhancement

area/model-registry

area/models