返回提交历史
Modified
g4f/Provider/hf/HuggingFaceAPI.py
+1
-6
XFEstudio/gpt4free
Fix pass session paramters in HuggingFaceAPI
eff81c30
代码差异
1 个文件
+1
-6
@@ -40,9 +40,6 @@ class HuggingFaceAPI(OpenaiTemplate):
40
40
api_base: str = None,
41
41
max_tokens: int = 2048,
42
42
max_inputs_lenght: int = 10000,
43
impersonate: str = None,
44
proxy: str = None,
45
timeout: int = 300,
46
43
images: ImagesType = None,
47
44
**kwargs
48
45
):
@@ -53,9 +50,7 @@ class HuggingFaceAPI(OpenaiTemplate):
53
50
api_base = f"https://api-inference.huggingface.co/models/{model_name}/v1"
54
51
if images is not None:
55
52
async with StreamSession(
56
proxy=proxy,
57
timeout=timeout,
58
impersonate=impersonate,
53
timeout=30,
59
54
) as session:
60
55
async with session.get(f"https://huggingface.co/api/models/{model}") as response:
61
56
if response.status == 404: