Skip to content

Commit 311807f

Browse files
albertvillanovaCyrilvallez
authored andcommitted
Remove invalid @staticmethod from module-level get_device_and_memory_breakdown (#41747)
Remove staticmethod decorator from function
1 parent 804038f commit 311807f

File tree

1 file changed

+0
-1
lines changed
  • src/transformers/generation/continuous_batching

1 file changed

+0
-1
lines changed

src/transformers/generation/continuous_batching/requests.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@
2828
# logger.setLevel(logging.INFO)
2929

3030

31-
@staticmethod
3231
def get_device_and_memory_breakdown() -> tuple[torch.device, int, int, int]:
3332
if torch.cuda.is_available():
3433
device = torch.device("cuda")

0 commit comments

Comments
 (0)