mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-06-04 11:24:00 +08:00
Add Llama3.3 (#4174)
### What problem does this PR solve? #4168 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
parent
cac7851fc5
commit
2cbe064080
@ -1014,6 +1014,18 @@
|
|||||||
"max_tokens": 131072,
|
"max_tokens": 131072,
|
||||||
"model_type": "chat"
|
"model_type": "chat"
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
"llm_name": "llama-3.3-70b-versatile",
|
||||||
|
"tags": "LLM,CHAT,128k",
|
||||||
|
"max_tokens": 128000,
|
||||||
|
"model_type": "chat"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"llm_name": "llama-3.3-70b-specdec",
|
||||||
|
"tags": "LLM,CHAT,8k",
|
||||||
|
"max_tokens": 8192,
|
||||||
|
"model_type": "chat"
|
||||||
|
},
|
||||||
{
|
{
|
||||||
"llm_name": "mixtral-8x7b-32768",
|
"llm_name": "mixtral-8x7b-32768",
|
||||||
"tags": "LLM,CHAT,5k",
|
"tags": "LLM,CHAT,5k",
|
||||||
|
@ -26,6 +26,7 @@ from api.utils.file_utils import get_project_base_directory
|
|||||||
from .operators import * # noqa: F403
|
from .operators import * # noqa: F403
|
||||||
from .operators import preprocess
|
from .operators import preprocess
|
||||||
|
|
||||||
|
|
||||||
class Recognizer(object):
|
class Recognizer(object):
|
||||||
def __init__(self, label_list, task_name, model_dir=None):
|
def __init__(self, label_list, task_name, model_dir=None):
|
||||||
"""
|
"""
|
||||||
@ -144,11 +145,11 @@ class Recognizer(object):
|
|||||||
return 0
|
return 0
|
||||||
x0_ = max(b["x0"], x0)
|
x0_ = max(b["x0"], x0)
|
||||||
x1_ = min(b["x1"], x1)
|
x1_ = min(b["x1"], x1)
|
||||||
assert x0_ <= x1_, "Fuckedup! T:{},B:{},X0:{},X1:{} ==> {}".format(
|
assert x0_ <= x1_, "Bbox mismatch! T:{},B:{},X0:{},X1:{} ==> {}".format(
|
||||||
tp, btm, x0, x1, b)
|
tp, btm, x0, x1, b)
|
||||||
tp_ = max(b["top"], tp)
|
tp_ = max(b["top"], tp)
|
||||||
btm_ = min(b["bottom"], btm)
|
btm_ = min(b["bottom"], btm)
|
||||||
assert tp_ <= btm_, "Fuckedup! T:{},B:{},X0:{},X1:{} => {}".format(
|
assert tp_ <= btm_, "Bbox mismatch! T:{},B:{},X0:{},X1:{} => {}".format(
|
||||||
tp, btm, x0, x1, b)
|
tp, btm, x0, x1, b)
|
||||||
ov = (btm_ - tp_) * (x1_ - x0_) if x1 - \
|
ov = (btm_ - tp_) * (x1_ - x0_) if x1 - \
|
||||||
x0 != 0 and btm - tp != 0 else 0
|
x0 != 0 and btm - tp != 0 else 0
|
||||||
|
@ -1,6 +1,5 @@
|
|||||||
import logging
|
import logging
|
||||||
import boto3
|
import boto3
|
||||||
import os
|
|
||||||
from botocore.exceptions import ClientError
|
from botocore.exceptions import ClientError
|
||||||
from botocore.client import Config
|
from botocore.client import Config
|
||||||
import time
|
import time
|
||||||
|
Loading…
x
Reference in New Issue
Block a user