silentchen commited on
Commit
f3f8f39
·
1 Parent(s): 74efa73

update space

Browse files
Files changed (1) hide show
  1. app.py +0 -1
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))