Spaces:
Runtime error
Runtime error
Commit
·
f3f8f39
1
Parent(s):
74efa73
update space
Browse files
app.py
CHANGED
|
@@ -158,7 +158,6 @@ def main():
|
|
| 158 |
# print(model)
|
| 159 |
|
| 160 |
# xm = load_model('transmitter', de
|
| 161 |
-
tmp_model = load_model('text300M', device=device)
|
| 162 |
with torch.no_grad():
|
| 163 |
new_proj = nn.Linear(1024 * 2, 1024, device=device, dtype=tmp_model.wrapped.input_proj.weight.dtype)
|
| 164 |
new_proj.weight = nn.Parameter(torch.zeros_like(new_proj.weight))
|
|
|
|
| 158 |
# print(model)
|
| 159 |
|
| 160 |
# xm = load_model('transmitter', de
|
|
|
|
| 161 |
with torch.no_grad():
|
| 162 |
new_proj = nn.Linear(1024 * 2, 1024, device=device, dtype=tmp_model.wrapped.input_proj.weight.dtype)
|
| 163 |
new_proj.weight = nn.Parameter(torch.zeros_like(new_proj.weight))
|