primerz commited on
Commit
fdb19bf
·
verified ·
1 Parent(s): b851544

Create config.py

Browse files
Files changed (1) hide show
  1. config.py +27 -0
config.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+
3
+ class Config:
4
+ # Hardware
5
+ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
6
+ DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
7
+
8
+ # Base Model & LoRA
9
+ REPO_ID = "primerz/pixagram"
10
+ CHECKPOINT_FILENAME = "horizon.safetensors"
11
+ LORA_FILENAME = "retroart.safetensors"
12
+
13
+ # Trigger Words
14
+ STYLE_TRIGGER = "p1x3l4rt, pixel art"
15
+
16
+ # InstantID Assets
17
+ INSTANTID_REPO = "InstantX/InstantID"
18
+ FACE_ADAPTER_REPO = "InstantX/InstantID"
19
+ FACE_ADAPTER_FILENAME = "ip-adapter.bin" # Usually auto-loaded by diffusers helper
20
+
21
+ # ControlNet Repos
22
+ CN_ZOE_REPO = "diffusers/controlnet-zoe-depth-sdxl-1.0"
23
+ CN_LINEART_REPO = "xinsir/controlnet-lineart-sdxl-1.0" # High quality lineart
24
+ CN_INSTANTID_REPO = "InstantX/InstantID" # Contains the ControlNet part
25
+
26
+ # Vision Models (for InsightFace)
27
+ ANTELOPEV2_PATH = "./models/antelopev2" # You might need to download this manually in utils