File size: 644 Bytes
23db765
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
"""Configuration constants for SpaceDebris Localizer."""

import os

MODEL_ID: str = os.getenv("MODEL_ID", "nvidia/LocateAnything-3B")
DEVICE: str = os.getenv("DEVICE", "cuda")
DTYPE: str = os.getenv("DTYPE", "bfloat16")
MAX_NEW_TOKENS: int = int(os.getenv("MAX_NEW_TOKENS", "8192"))
GENERATION_MODE: str = os.getenv("GENERATION_MODE", "hybrid")
TEMPERATURE: float = float(os.getenv("TEMPERATURE", "0.7"))
COORD_MAX: int = 1000
DEFAULT_CONFIDENCE: float = 0.85
APP_TITLE: str = "SpaceDebris Localizer"
APP_SUBTITLE: str = (
    "Use LocateAnything-3B to ground debris, satellite fragments, "
    "and spacecraft components in space imagery."
)