File size: 1,206 Bytes
338d95d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
"""
Configuration settings for CompI project.
"""
import os
from pathlib import Path
# Project paths
PROJECT_ROOT = Path(__file__).parent.parent
DATA_DIR = PROJECT_ROOT / "data"
OUTPUTS_DIR = PROJECT_ROOT / "outputs"
MODELS_DIR = PROJECT_ROOT / "models"
# Create directories if they don't exist
for dir_path in [DATA_DIR, OUTPUTS_DIR, MODELS_DIR]:
dir_path.mkdir(exist_ok=True)
# Model configurations
STABLE_DIFFUSION_MODEL = "runwayml/stable-diffusion-v1-5"
STABLE_DIFFUSION_IMG2IMG_MODEL = "runwayml/stable-diffusion-v1-5"
EMOTION_MODEL = "j-hartmann/emotion-english-distilroberta-base"
SENTIMENT_MODEL = "cardiffnlp/twitter-roberta-base-sentiment-latest"
# Generation settings
DEFAULT_IMAGE_SIZE = (512, 512)
DEFAULT_INFERENCE_STEPS = 20
DEFAULT_GUIDANCE_SCALE = 7.5
# Audio settings
SAMPLE_RATE = 22050
AUDIO_DURATION = 10 # seconds
# Device settings
DEVICE = "cuda" if os.getenv("CUDA_AVAILABLE", "false").lower() == "true" else "cpu"
# API keys (load from environment)
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
# Logging
LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO")
LOG_FORMAT = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|