Spaces:
Sleeping
Sleeping
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()
|