Update README.md
Browse files
README.md
CHANGED
|
@@ -104,7 +104,6 @@ model.to(device)
|
|
| 104 |
|
| 105 |
out = processor(audios=audios, texts=texts)
|
| 106 |
out = {k: v.to(device) if torch.is_tensor(v) else v for k, v in out.items()}
|
| 107 |
-
out = dict(list(out.items())[:-2])
|
| 108 |
|
| 109 |
with torch.no_grad() :
|
| 110 |
preds = model(**out).logits
|
|
|
|
| 104 |
|
| 105 |
out = processor(audios=audios, texts=texts)
|
| 106 |
out = {k: v.to(device) if torch.is_tensor(v) else v for k, v in out.items()}
|
|
|
|
| 107 |
|
| 108 |
with torch.no_grad() :
|
| 109 |
preds = model(**out).logits
|