Mirror/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2024-06-07 19:40:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove annotations and add Enum

Browse Source 
Signed-off-by: GitHub <noreply@github.com>
This commit is contained in:
Aisuko 2023-11-09 00:54:25 +00:00 committed by GitHub
parent 778054f234
commit 63bd7a5674
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
extra/grpc/sam/sam.py
View File

@ -1,8 +1,8 @@
#! /usr/bin/env python3 #! /usr/bin/env python3
from __future__ import annotations
from concurrent import futures from concurrent import futures
import argparse import argparse
from enum import Enum
import os import os
import signal import signal
import sys import sys
@ -28,7 +28,7 @@ IMAGE_SIZE = 1024
VIT_PATCH_SIZE=16 VIT_PATCH_SIZE=16
# Enum for sam model type # Enum for sam model type
class SamModelType: class SamModelType(str, Enum):
default = "sam_hq_vit_h.pth" default = "sam_hq_vit_h.pth"
vit_h = "sam_hq_vit_h.pth" vit_h = "sam_hq_vit_h.pth"
vit_l = "sam_hq_vit_l.pth" vit_l = "sam_hq_vit_l.pth"