Update README.md
Browse files
README.md
CHANGED
|
@@ -36,8 +36,6 @@ with torch.no_grad():
|
|
| 36 |
|
| 37 |
pixel_values = processor(img, random_padding=False, return_tensors="pt").pixel_values
|
| 38 |
batch_size = pixel_values.shape[0]
|
| 39 |
-
# In PyTorch DDP setup, .model is wrapped by DistributedDataParallel,
|
| 40 |
-
# so the actual model is accessible through .model.module.
|
| 41 |
decoder_input_ids = torch.full((batch_size, 1), model.config.decoder_start_token_id,
|
| 42 |
device=device)
|
| 43 |
|
|
|
|
| 36 |
|
| 37 |
pixel_values = processor(img, random_padding=False, return_tensors="pt").pixel_values
|
| 38 |
batch_size = pixel_values.shape[0]
|
|
|
|
|
|
|
| 39 |
decoder_input_ids = torch.full((batch_size, 1), model.config.decoder_start_token_id,
|
| 40 |
device=device)
|
| 41 |
|