Great, now that attention slicing is enabled, let’s try to double the batch size again, going for batch_size=8.                                                                                                                              