Update README.md
Browse files
README.md
CHANGED
|
@@ -87,7 +87,6 @@ tgt_output = model(tgt=inputs)["tgt_reps"]
|
|
| 87 |
print(string, '=', model.compute_similarity(qry_output, tgt_output))
|
| 88 |
|
| 89 |
# Text -> Image
|
| 90 |
-
|
| 91 |
inputs = processor('Find me an everyday image that matches the given caption: A cat and a dog.',)
|
| 92 |
inputs = {key: value.to('cuda') for key, value in inputs.items()}
|
| 93 |
qry_output = model(qry=inputs)["qry_reps"]
|
|
|
|
| 87 |
print(string, '=', model.compute_similarity(qry_output, tgt_output))
|
| 88 |
|
| 89 |
# Text -> Image
|
|
|
|
| 90 |
inputs = processor('Find me an everyday image that matches the given caption: A cat and a dog.',)
|
| 91 |
inputs = {key: value.to('cuda') for key, value in inputs.items()}
|
| 92 |
qry_output = model(qry=inputs)["qry_reps"]
|