Update src/util/params.py

src/util/params.py

@@ -15,7 +15,7 @@ torch_device = "cuda" if torch.cuda.is_available() else "cpu"
 isLCM = False
 HF_ACCESS_TOKEN = ""
 
-model_path = "stabilityai/stable-diffusion-2-1"
+model_path = "segmind/small-sd"
 inpaint_model_path = "Lykon/dreamshaper-8-inpainting"
 prompt = "Self-portrait oil painting, a beautiful cyborg with golden hair, 8k"
 promptA = "Self-portrait oil painting, a beautiful man with golden hair, 8k"
@@ -33,7 +33,7 @@ guidance_values = "1, 8, 20"
 intermediate = True
 pokeX, pokeY = 256, 256
 pokeHeight, pokeWidth = 128, 128
-imageHeight, imageWidth = 1024, 1024
+imageHeight, imageWidth = 512, 512
 
 tokenizer = CLIPTokenizer.from_pretrained(model_path, subfolder="tokenizer")
 text_encoder = CLIPTextModel.from_pretrained(model_path, subfolder="text_encoder").to(