Fix remaining tool loading issues

- Remove BitsAndBytesConfig import from NV-Reason-CXR (causing dict.to_dict() error)
- Remove load_in_4bit parameter from Classification and Report Generation tools
- Update timm to 0.9.16 to fix ImageNetInfo import error for VQA tool
- All tools should now load properly

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

app.py

@@ -72,8 +72,7 @@ if device == "cuda":
     try:
         from medrax.tools.classification import TorchXRayVisionClassifierTool
         classification_tool = TorchXRayVisionClassifierTool(
-            device=device,
-            load_in_4bit=True
+            device=device
         )
         tools.append(classification_tool)
         print("✓ Loaded classification tool")
@@ -83,8 +82,7 @@ if device == "cuda":
     try:
         from medrax.tools.report_generation import ChestXRayReportGeneratorTool
         report_tool = ChestXRayReportGeneratorTool(
-            device=device,
-            load_in_4bit=True
+            device=device
         )
         tools.append(report_tool)
         print("✓ Loaded report generation tool")

medrax/tools/nv_reason_cxr.py

@@ -4,7 +4,7 @@ from pathlib import Path
 import torch
 from PIL import Image
 from pydantic import BaseModel, Field
-from transformers import AutoProcessor, AutoModelForImageTextToText, BitsAndBytesConfig
+from transformers import AutoProcessor, AutoModelForImageTextToText
 
 from langchain_core.callbacks import (
     AsyncCallbackManagerForToolRun,

requirements.txt

@@ -41,7 +41,7 @@ fastapi>=0.68.0
 python-multipart>=0.0.6
 einops>=0.3.0
 einops-exts>=0.0.4
-timm==0.5.4
+timm==0.9.16
 tiktoken>=0.3.0
 openai>=0.27.0
 backoff>=1.10.0