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))
|