Add `phi-1_5` model (#493)

NOTE: Currently only works with node.js backend
This commit is contained in:
Joshua Lochner 2024-01-02 23:12:09 +02:00 committed by GitHub
parent 38ec1d41b3
commit 81f1e5c320
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 0 deletions

View File

@ -75,6 +75,10 @@ MODEL_SPECIFIC_QUANTIZE_PARAMS = {
'per_channel': False,
'reduce_range': False,
},
'phi': {
'per_channel': False,
'reduce_range': False,
},
# Encoder-decoder models
'whisper': {

View File

@ -714,6 +714,7 @@ SUPPORTED_MODELS = {
# Text generation
'text-generation': [
'hf-internal-testing/tiny-random-PhiForCausalLM',
'susnato/phi-1_5_dev',
],
},
'roberta': {