Add `phi-1_5` model (#493)
NOTE: Currently only works with node.js backend
This commit is contained in:
parent
38ec1d41b3
commit
81f1e5c320
|
@ -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': {
|
||||
|
|
|
@ -714,6 +714,7 @@ SUPPORTED_MODELS = {
|
|||
# Text generation
|
||||
'text-generation': [
|
||||
'hf-internal-testing/tiny-random-PhiForCausalLM',
|
||||
'susnato/phi-1_5_dev',
|
||||
],
|
||||
},
|
||||
'roberta': {
|
||||
|
|
Loading…
Reference in New Issue