Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Fireworks AI provider + instructions for new provider #2848

Merged
merged 45 commits into from
Feb 12, 2025
Merged
Changes from 1 commit
Commits
Show all changes
45 commits
Select commit Hold shift + click to select a range
dbda294
first draft of dynamic mapping
hanouticelina Feb 5, 2025
2c56667
fix imports and typing
hanouticelina Feb 5, 2025
2760a76
add back recommended models fetching for hf-inference
hanouticelina Feb 5, 2025
7783943
fix
hanouticelina Feb 5, 2025
cc73ead
avoir circular imports
hanouticelina Feb 5, 2025
a178f03
small clean up
hanouticelina Feb 5, 2025
3956df5
add default supported model list
hanouticelina Feb 5, 2025
f2208ae
remove unnecessary arg
hanouticelina Feb 5, 2025
d43462a
nit
hanouticelina Feb 5, 2025
4988db0
rename function
hanouticelina Feb 5, 2025
91f41c3
another nit
hanouticelina Feb 5, 2025
cb7eb9a
fix
hanouticelina Feb 5, 2025
8d78dca
fix conversational
hanouticelina Feb 5, 2025
181b6a7
fix hf-inference
hanouticelina Feb 5, 2025
0568aa5
add warning when status=staging
hanouticelina Feb 5, 2025
d71e92e
update warning and use model_info
hanouticelina Feb 5, 2025
cf1e956
update import
hanouticelina Feb 5, 2025
949e8c6
fix ExpandModelProperty_T
hanouticelina Feb 5, 2025
80acc5c
Merge branch 'main' into add-dynamic-inference-provider-mapping
Wauplin Feb 7, 2025
5639de0
refacto
Wauplin Feb 7, 2025
53cca3c
fix python 3.8
Wauplin Feb 7, 2025
0474833
fix test
Wauplin Feb 10, 2025
ee5443a
remove newlines
Wauplin Feb 10, 2025
5dfe4b9
Base class for inference providers
Wauplin Feb 10, 2025
faf19e0
revert
Wauplin Feb 10, 2025
632d43e
refacto hf-inference and fal-ai tests
Wauplin Feb 10, 2025
4053a15
replicate tests
Wauplin Feb 10, 2025
90521c4
samba and together tests
Wauplin Feb 10, 2025
72396a8
reorder
Wauplin Feb 10, 2025
c6ec0f8
unfinished business
Wauplin Feb 10, 2025
6969309
some docstrings
Wauplin Feb 10, 2025
0c39884
fix some tests
Wauplin Feb 10, 2025
449342d
fix HfInference does not require token
Wauplin Feb 10, 2025
bd8c8e4
fix inference client tests
hanouticelina Feb 11, 2025
9d6f2a3
Merge branch 'main' into add-dynamic-inference-provider-mapping
Wauplin Feb 11, 2025
d041ceb
fix hf-inference _prepare_api_key
Wauplin Feb 11, 2025
e8083c0
fai ai get response tests
Wauplin Feb 11, 2025
eebadd4
test get_response together + replicate
Wauplin Feb 11, 2025
38066e0
fix prepare_url
Wauplin Feb 11, 2025
2f06b9d
Add Fireworks AI provider + instructions for new provider
Wauplin Feb 11, 2025
b97fea3
Update src/huggingface_hub/inference/_providers/new_provider.md
Wauplin Feb 11, 2025
4324dea
add fireworks AI to supported providers table
Wauplin Feb 11, 2025
5a48909
Merge branch 'add-fireworks-provider' of github.com:huggingface/huggi…
Wauplin Feb 11, 2025
b1c7693
Merge branch 'main' into add-fireworks-provider
Wauplin Feb 11, 2025
b9f132d
Merge branch 'main' into add-fireworks-provider
Wauplin Feb 12, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
update import
hanouticelina committed Feb 5, 2025
commit cf1e9562480e0aa714c6ed163355271fee214dc8
2 changes: 1 addition & 1 deletion src/huggingface_hub/inference/_common.py
Original file line number Diff line number Diff line change
@@ -112,7 +112,7 @@ def _fetch_provider_mappings(model: str) -> Dict:
"""
Fetch provider mappings for a model from the Hub.
"""
from huggingface_hub.hf_api import model_info
from ..hf_api import model_info

info = model_info(model, expand=["inferenceProviderMapping"])
provider_mapping = info.inference_provider_mapping