msy127 commited on
Commit
d7cb645
·
1 Parent(s): ea5ba62

Update app.py

Browse files

202311011-1229

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -30,6 +30,7 @@ class KwArgsModel(BaseModel):
30
  kwargs: Dict[str, Any] = Field(default_factory=dict)
31
 
32
  class CustomInferenceClient(LLM, KwArgsModel):
 
33
  model_name: str
34
  inference_client: InferenceClient
35
 
 
30
  kwargs: Dict[str, Any] = Field(default_factory=dict)
31
 
32
  class CustomInferenceClient(LLM, KwArgsModel):
33
+ __metaclass__ = type(LLM)
34
  model_name: str
35
  inference_client: InferenceClient
36