hiddenFront commited on
Commit
58648d1
ยท
verified ยท
1 Parent(s): 95c8334

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -126,7 +126,10 @@ try:
126
  name = name[7:]
127
  new_state_dict[name] = v
128
 
129
- model.load_state_dict(new_state_dict)
 
 
 
130
  # --- ์ˆ˜์ •๋œ ํ•ต์‹ฌ ๋ถ€๋ถ„ ๋ ---
131
 
132
  model.to(device) # ๋ชจ๋ธ์„ ๋””๋ฐ”์ด์Šค๋กœ ์ด๋™
 
126
  name = name[7:]
127
  new_state_dict[name] = v
128
 
129
+ # strict=False๋ฅผ ์‚ฌ์šฉํ•˜์—ฌ Missing key(s) ์˜ค๋ฅ˜๋ฅผ ๋ฐฉ์ง€ํ•ฉ๋‹ˆ๋‹ค.
130
+ # ์ด๋Š” new_state_dict์— ์—†๋Š” ํ‚ค๋Š” ๋ชจ๋ธ์—์„œ ๊ธฐ์กด ๊ฐ’(from_pretrained๋กœ ๋กœ๋“œ๋œ)์„ ์œ ์ง€ํ•˜๊ณ ,
131
+ # ๋ชจ๋ธ์— ์—†๋Š” ํ‚ค๋Š” ๋ฌด์‹œํ•˜๋„๋ก ํ•ฉ๋‹ˆ๋‹ค.
132
+ model.load_state_dict(new_state_dict, strict=False)
133
  # --- ์ˆ˜์ •๋œ ํ•ต์‹ฌ ๋ถ€๋ถ„ ๋ ---
134
 
135
  model.to(device) # ๋ชจ๋ธ์„ ๋””๋ฐ”์ด์Šค๋กœ ์ด๋™