Anew511/extras

1# Constants2DEFAULT_CUDA_DEVICE = "cuda:0"3# Also try: 'Qiliang/bart-large-cnn-samsum-ElectrifAi_v10'4DEFAULT_SUMMARIZATION_MODEL = "Qiliang/bart-large-cnn-samsum-ChatGPT_v3"5# Also try: 'joeddav/distilbert-base-uncased-go-emotions-student'6DEFAULT_CLASSIFICATION_MODEL = "nateraw/bert-base-uncased-emotion"7# Also try: 'Salesforce/blip-image-captioning-base'8DEFAULT_CAPTIONING_MODEL = "Salesforce/blip-image-captioning-large"9DEFAULT_SD_MODEL = "ckpt/anything-v4.5-vae-swapped"10DEFAULT_EMBEDDING_MODEL = "sentence-transformers/all-mpnet-base-v2"11DEFAULT_REMOTE_SD_HOST = "127.0.0.1"12DEFAULT_REMOTE_SD_PORT = 786013DEFAULT_CHROMA_PORT = 800014SILERO_SAMPLES_PATH = "tts_samples"15SILERO_SAMPLE_TEXT = "The quick brown fox jumps over the lazy dog"16# ALL_MODULES = ['caption', 'summarize', 'classify', 'keywords', 'prompt', 'sd']17DEFAULT_SUMMARIZE_PARAMS = {18    "temperature": 1.0,19    "repetition_penalty": 1.0,20    "max_length": 500,21    "min_length": 200,22    "length_penalty": 1.5,23    "bad_words": [24        "\n",25        '"',26        "*",27        "[",28        "]",29        "{",30        "}",31        ":",32        "(",33        ")",34        "<",35        ">",36        "Â",37        "The text ends",38        "The story ends",39        "The text is",40        "The story is",41    ],42}43 44PROMPT_PREFIX = "best quality, absurdres, "45NEGATIVE_PROMPT = """lowres, bad anatomy, error body, error hair, error arm,46error hands, bad hands, error fingers, bad fingers, missing fingers47error legs, bad legs, multiple legs, missing legs, error lighting,48error shadow, error reflection, text, error, extra digit, fewer digits,49cropped, worst quality, low quality, normal quality, jpeg artifacts,50signature, watermark, username, blurry"""51