Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -126,7 +126,10 @@ try:
|
|
| 126 |
name = name[7:]
|
| 127 |
new_state_dict[name] = v
|
| 128 |
|
| 129 |
-
|
|
|
|
|
|
|
|
|
|
| 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) # ๋ชจ๋ธ์ ๋๋ฐ์ด์ค๋ก ์ด๋
|