config.py

import torch

# Model configurations
MODEL_NAME = "gpt2-medium"
SPEECH_MODEL = "facebook/wav2vec2-base-960h"
TTS_MODEL = "v3_en"

# Device configuration
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

# Memory configurations
MEMORY_SIZE = 10 * 1024 * 1024 * 1024  # 10GB in bytes
LOCAL_MEMORY_SIZE = 3 * 1024 * 1024 * 1024  # 3GB in bytes

# API configurations
API_KEY = "your_huggingface_api_key_here"

# Conversation settings
MAX_CONTEXT_LENGTH = 3024