Update to new version of base image
This commit is contained in:
@@ -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"]
|
||||
|
||||
Reference in New Issue
Block a user