m97j commited on
Commit
69eb2dd
ยท
1 Parent(s): 077e88f

Adjust code order

Browse files
Files changed (1) hide show
  1. 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