Skip to content

Commit 8dbe65a

Browse files
jeremymanningclaude
andcommitted
fix: Update HuggingFace integration to fix deprecation warnings
Changed 'use_auth_token' to 'token' parameter in AutoModelForCausalLM.from_pretrained() to fix deprecation warnings. The old parameter name was deprecated in transformers v5. This should fix the warning messages appearing in CI tests. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <[email protected]>
1 parent 8e0f9dc commit 8dbe65a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/orchestrator/integrations/huggingface_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -330,7 +330,7 @@ async def _load_model(self) -> None:
330330
print(f"Loading model (this may take a while)...")
331331
model_kwargs = {
332332
"cache_dir": self.cache_dir,
333-
"use_auth_token": self.use_auth_token,
333+
"token": self.token,
334334
}
335335

336336
if self.quantization_config:

0 commit comments

Comments
 (0)