vectorplasticity commited on
Commit
c6924d7
·
verified ·
1 Parent(s): b36be34

Add missing DEFAULT_HARDWARE and AVAILABLE_HARDWARE settings

Browse files
Files changed (1) hide show
  1. 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 = {