Update 1/model.py
This commit is contained in:
parent
d96a00a896
commit
df02a952af
@ -3,6 +3,7 @@ import torch
|
|||||||
import numpy as np
|
import numpy as np
|
||||||
import triton_python_backend_utils as pb_utils
|
import triton_python_backend_utils as pb_utils
|
||||||
import uuid
|
import uuid
|
||||||
|
import transformers
|
||||||
from typing import List, Dict, Any, Union, Tuple
|
from typing import List, Dict, Any, Union, Tuple
|
||||||
|
|
||||||
from transformers import (
|
from transformers import (
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user