File size: 193 Bytes
70de1d6
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from pipeline import Pipeline
from device import device, torch_dtype

def main():
    p = Pipeline(device, torch_dtype)
    print(p.InputParams.schema())

if __name__ == "__main__":
    main()