pixel-art / config.py
primerz's picture
Update config.py
006a6f3 verified
raw
history blame
762 Bytes
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