NurseCitizenDeveloper commited on
Commit
c856552
·
verified ·
1 Parent(s): b3d2e15

Update carebridge_client.py

Browse files
Files changed (1) hide show
  1. carebridge_client.py +1 -1
carebridge_client.py CHANGED
@@ -17,7 +17,7 @@ class CareBridgeTranslator:
17
 
18
  def _load_model(self):
19
  if self.model is None:
20
- print(f"[SIMBOTI] Loading model ${self.model_id}...")
21
  self.processor = AutoProcessor.from_pretrained(self.model_id)
22
  self.model = AutoModelForImageTextToText.from_pretrained(self.model_id, device_map="cuda", torch_dtype=torch.float16)
23
  print("[SIMBOTI] Model loaded successfully.")
 
17
 
18
  def _load_model(self):
19
  if self.model is None:
20
+ print(f"[SIMBOTI] Loading model {self.model_id}...")
21
  self.processor = AutoProcessor.from_pretrained(self.model_id)
22
  self.model = AutoModelForImageTextToText.from_pretrained(self.model_id, device_map="cuda", torch_dtype=torch.float16)
23
  print("[SIMBOTI] Model loaded successfully.")