transformers_tasks
transformers_tasks copied to clipboard
打开标注平台的时候出现以下bug
not supported between instances of 'str' and 'int

@HarderThenHarder
Hi,看起来像是版本不同导致device
格式不一致导致的。
您尝试下把 代码的这里 做如下修改:
...
device=MODEL_CONFIG['device'] -> device=0
传入一个 int
类型的 GPU 索引,索引决定了模型在进行内容生成的时候具体使用哪块 GPU。
我打开平台,会出这个bug
NameError: name 'model' is not defined