diff --git a/config.py b/config.py index 268dfc9..3c6435d 100644 --- a/config.py +++ b/config.py @@ -5,10 +5,12 @@ "Version": "0.1.0", "Author": "DeepMake", "Description": "Stable Diffusion generation using the Diffusers library", - "env": "sd" + "env": "sd", + "memory": 4000, + "model_memory": 3000 } config = { - "model_name": "runwayml/stable-diffusion-v1-5", + "model_name": "runwayml/stable-diffusion-v1-5", # "options": ["runwayml/stable-diffusion-v1-5", "stabilityai/sdxl-turbo"]}, "model_dtype": "fp32" if sys.platform == "darwin" else "fp16", "loras": [], "inverters": [],