Add missing DEFAULT_HARDWARE and AVAILABLE_HARDWARE settings
Browse files- app/config.py +2 -0
app/config.py
CHANGED
|
@@ -52,6 +52,8 @@ class Settings(BaseSettings):
|
|
| 52 |
# Hardware
|
| 53 |
DEVICE: str = "auto" # auto, cpu, cuda, mps
|
| 54 |
DTYPE: str = "auto" # auto, float32, float16, bfloat16
|
|
|
|
|
|
|
| 55 |
|
| 56 |
# PEFT / LoRA Defaults
|
| 57 |
PEFT_DEFAULTS: dict = {
|
|
|
|
| 52 |
# Hardware
|
| 53 |
DEVICE: str = "auto" # auto, cpu, cuda, mps
|
| 54 |
DTYPE: str = "auto" # auto, float32, float16, bfloat16
|
| 55 |
+
DEFAULT_HARDWARE: str = "auto"
|
| 56 |
+
AVAILABLE_HARDWARE: List[str] = ["cpu", "cuda", "auto"]
|
| 57 |
|
| 58 |
# PEFT / LoRA Defaults
|
| 59 |
PEFT_DEFAULTS: dict = {
|