Fix installed torch version to be compatible with cuda drivers
This commit is contained in:
@@ -320,3 +320,6 @@ def predict(
|
||||
writer.close()
|
||||
|
||||
typer.echo(f"wrote {total:,} rows → {out}")
|
||||
|
||||
if __name__ == "__main__":
|
||||
app()
|
||||
|
||||
Reference in New Issue
Block a user