vaibhavm29 commited on
Commit
99c1e69
Β·
1 Parent(s): 7d4bd7e

removed excess lines

Browse files
Files changed (1) hide show
  1. medrax/tools/xray_vqa.py +0 -1
medrax/tools/xray_vqa.py CHANGED
@@ -46,7 +46,6 @@ class XRayVQATool(BaseTool):
46
  def __init__(
47
  self,
48
  model_name: str = "StanfordAIMI/CheXagent-2-3b",
49
- # model_name: str = "google/medgemma-4b-it",
50
  device: Optional[str] = "cuda",
51
  dtype: torch.dtype = torch.bfloat16,
52
  cache_dir: Optional[str] = None,
 
46
  def __init__(
47
  self,
48
  model_name: str = "StanfordAIMI/CheXagent-2-3b",
 
49
  device: Optional[str] = "cuda",
50
  dtype: torch.dtype = torch.bfloat16,
51
  cache_dir: Optional[str] = None,