Spaces:
Running
on
T4
Running
on
T4
FantasticGu
commited on
Commit
•
f0a1818
1
Parent(s):
00f5723
update
Browse files- model/openllama.py +1 -1
model/openllama.py
CHANGED
|
@@ -230,7 +230,7 @@ class OpenLLAMAPEFTModel(nn.Module):
|
|
| 230 |
|
| 231 |
self.llama_proj = nn.Linear(
|
| 232 |
self.visual_hidden_size, self.llama_model.config.hidden_size
|
| 233 |
-
)
|
| 234 |
|
| 235 |
self.max_tgt_len = max_tgt_len
|
| 236 |
|
|
|
|
| 230 |
|
| 231 |
self.llama_proj = nn.Linear(
|
| 232 |
self.visual_hidden_size, self.llama_model.config.hidden_size
|
| 233 |
+
).to(self.device)
|
| 234 |
|
| 235 |
self.max_tgt_len = max_tgt_len
|
| 236 |
|