Case-Study-1 / config.py
Julian-Hans's picture
renamed poc_app.py to app.py, clean up for app logic, implemented function to yield intermediate results, added generation length parameter to config, changed path handling in app.py
ddf2ccc
raw
history blame
283 Bytes
IMAGE_CAPTION_MODEL = "Salesforce/blip-image-captioning-large"
LLM_MODEL = "microsoft/Phi-3-mini-4k-instruct"
LLM_MAX_LENGTH = 50
LLM_MAX_NEW_TOKENS = 50
MUSICGEN_MODEL = "facebook/musicgen-small"
MUSICGEN_MAX_NEW_TOKENS = 256 # 5 seconds of audio
AUDIO_DIR = "Case-Study-1/data/"