File size: 762 Bytes
6026000
 
 
 
 
 
 
006a6f3
 
 
6026000
006a6f3
6026000
 
 
 
006a6f3
6026000
 
 
006a6f3
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import torch

class Config:
    # Hardware
    DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
    DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32

    # Base Model & LoRA (from primerz/pixagram)
    REPO_ID = "primerz/pixagram"
    CHECKPOINT_FILENAME = "horizon.safetensors"
    LORA_FILENAME = "retroart.safetensors"
    LORA_STRENGTH = 1.2 
    
    # Trigger Words for the LoRA
    STYLE_TRIGGER = "p1x3l4rt, pixel art"
    
    # Default Negative Prompt
    DEFAULT_NEGATIVE_PROMPT = "Ugly, artifacts, blurry, disformed, photo-realistic, photo, photography, realistic, low-quality, pixelart, text."

    # Gradio Parameters
    CGF_SCALE = 1.2           
    STEPS_NUMBER = 8          # Reduced slightly for optimal LCM T2I performance