liguang0115's picture
Remove redundant import of torch in app.py and add it back for proper functionality with GPU context and autocasting.
28b109f