suayptalha commited on
Commit
b8c2306
·
verified ·
1 Parent(s): 10ad461

Update modeling_minGRULM.py

Browse files
Files changed (1) hide show
  1. modeling_minGRULM.py +1 -0
modeling_minGRULM.py CHANGED
@@ -137,6 +137,7 @@ class MinGRULMForCausalLM(PreTrainedModel):
137
  save_directory (str): Directory to save the model.
138
  safe_serialization (bool, optional): Whether to use safe serialization. Defaults to True.
139
  """
 
140
  os.makedirs(save_directory, exist_ok=True)
141
 
142
  if safe_serialization:
 
137
  save_directory (str): Directory to save the model.
138
  safe_serialization (bool, optional): Whether to use safe serialization. Defaults to True.
139
  """
140
+ import os
141
  os.makedirs(save_directory, exist_ok=True)
142
 
143
  if safe_serialization: