File tree Expand file tree Collapse file tree
customize/router-builder/task-router Expand file tree Collapse file tree Original file line number Diff line number Diff line change 430430 " tokenizer = AutoTokenizer.from_pretrained(\" nvidia/prompt-task-and-complexity-classifier\" )\n " ,
431431 " \n " ,
432432 " # Load the traced model\n " ,
433- " wrapped_model = torch.jit.load('prompt_classifier_traced .pt')\n " ,
433+ " wrapped_model = torch.jit.load('triton_template/task_router/1/model .pt')\n " ,
434434 " wrapped_model.eval()\n " ,
435435 " \n " ,
436436 " # Prepare a sample input\n " ,
Original file line number Diff line number Diff line change 478478 "metadata": {},
479479 "outputs": [],
480480 "source": [
481- "!cd .. && make up"
481+ "!cd .. && make up > /dev/null 2>&1 "
482482 ]
483483 },
484484 {
789789 "metadata": {},
790790 "outputs": [],
791791 "source": [
792- "!cd .. && make app"
792+ "!cd .. && make app > /dev/null 2>&1 "
793793 ]
794794 },
795795 {
You can’t perform that action at this time.
0 commit comments