Update to new version of base image

vllm/lora/models.py

@@ -2,7 +2,7 @@ import copy
 import json
 import math
 import os
-import re
+import regex as re
 from dataclasses import dataclass, field
 from typing import Any, Callable, Dict, List, Optional, Type
 
@@ -263,7 +263,7 @@ class LoRAModel(AdapterModel):
                 new_embeddings_tensor_path)
         elif os.path.isfile(new_embeddings_bin_file_path):
             embeddings = torch.load(new_embeddings_bin_file_path,
-                                    map_location=device)
+                                    map_location=device, weights_only=True)
 
         rank = config["r"]
         lora_alpha = config["lora_alpha"]

vllm/lora/utils.py

@@ -1,5 +1,5 @@
 import os
-import re
+import regex as re
 from typing import List, Optional, Set, Tuple, Type, Union
 
 import huggingface_hub