Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
3392c51
1
Parent(s):
85ea87c
Update utils/common_viz.py
Browse files- utils/common_viz.py +0 -1
utils/common_viz.py
CHANGED
|
@@ -88,7 +88,6 @@ def get_batch(
|
|
| 88 |
batch = collate_fn([to_device(raw_batch, device)])
|
| 89 |
|
| 90 |
# Encode text
|
| 91 |
-
clip_model.to(device)
|
| 92 |
caption_seq, caption_tokens = encode_text([prompt], clip_model, None, device)
|
| 93 |
print(caption_seq[0].device)
|
| 94 |
|
|
|
|
| 88 |
batch = collate_fn([to_device(raw_batch, device)])
|
| 89 |
|
| 90 |
# Encode text
|
|
|
|
| 91 |
caption_seq, caption_tokens = encode_text([prompt], clip_model, None, device)
|
| 92 |
print(caption_seq[0].device)
|
| 93 |
|