Spaces:
Running
Running
Adjust code order
Browse files- model_loader.py +2 -1
model_loader.py
CHANGED
@@ -36,6 +36,8 @@ class ModelWrapper:
|
|
36 |
token=HF_TOKEN
|
37 |
)
|
38 |
|
|
|
|
|
39 |
# LoRA ์ด๋ํฐ ์ ์ฉ
|
40 |
self.model = PeftModel.from_pretrained(
|
41 |
base,
|
@@ -45,7 +47,6 @@ class ModelWrapper:
|
|
45 |
token=HF_TOKEN
|
46 |
)
|
47 |
|
48 |
-
self.model.resize_token_embeddings(len(self.tokenizer))
|
49 |
|
50 |
# ์ปค์คํ
ํค๋ ์ถ๊ฐ
|
51 |
hidden_size = self.model.config.hidden_size
|
|
|
36 |
token=HF_TOKEN
|
37 |
)
|
38 |
|
39 |
+
self.model.resize_token_embeddings(len(self.tokenizer))
|
40 |
+
|
41 |
# LoRA ์ด๋ํฐ ์ ์ฉ
|
42 |
self.model = PeftModel.from_pretrained(
|
43 |
base,
|
|
|
47 |
token=HF_TOKEN
|
48 |
)
|
49 |
|
|
|
50 |
|
51 |
# ์ปค์คํ
ํค๋ ์ถ๊ฐ
|
52 |
hidden_size = self.model.config.hidden_size
|