Ahmet Kaan Sever commited on
Commit
f05ebc2
·
1 Parent(s): 631bbc7

added debug print lines

Browse files
Files changed (1) hide show
  1. src/deepeval/base_task.py +2 -0
src/deepeval/base_task.py CHANGED
@@ -28,12 +28,14 @@ class BaseTask(ABC):
28
  @staticmethod
29
  def load_model(model_name: str, device):
30
  """Loads model and tokenizer once and caches it."""
 
31
  model = AutoModelForCausalLM.from_pretrained(
32
  model_name,
33
  torch_dtype=torch.float16,
34
  device_map=device,
35
  token=HF_TOKEN, # Replace with actual token
36
  )
 
37
  tokenizer = AutoTokenizer.from_pretrained(model_name)
38
  return model, tokenizer
39
 
 
28
  @staticmethod
29
  def load_model(model_name: str, device):
30
  """Loads model and tokenizer once and caches it."""
31
+ print(f"Loading model: {model_name}")
32
  model = AutoModelForCausalLM.from_pretrained(
33
  model_name,
34
  torch_dtype=torch.float16,
35
  device_map=device,
36
  token=HF_TOKEN, # Replace with actual token
37
  )
38
+ print("Model loaded.")
39
  tokenizer = AutoTokenizer.from_pretrained(model_name)
40
  return model, tokenizer
41