mirror of
https://github.com/immich-app/immich.git
synced 2025-01-21 00:52:43 -05:00
6 lines
195 B
Python
6 lines
195 B
Python
|
from app.schemas import ModelSession
|
||
|
|
|
||
|
|
|
||
|
|
def has_batch_axis(session: ModelSession) -> bool:
|
||
|
|
return not isinstance(session.get_inputs()[0].shape[0], int) or session.get_inputs()[0].shape[0] < 0
|