Upload cli-sd15-editing.py with huggingface_hub
Browse files- cli-sd15-editing.py +1 -2
cli-sd15-editing.py
CHANGED
|
@@ -59,8 +59,7 @@ def main(args):
|
|
| 59 |
# Model
|
| 60 |
disable_torch_init()
|
| 61 |
|
| 62 |
-
|
| 63 |
-
model_name = 'llmga'
|
| 64 |
tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, args.model_base, model_name, args.load_8bit, args.load_4bit, device=args.device)
|
| 65 |
vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
|
| 66 |
pipe = SemanticStableDiffusionImg2ImgPipeline_DPMSolver.from_pretrained(args.sd_model_id,vae=vae,torch_dtype=torch.float16, safety_checker=None, requires_safety_checker=False).to("cuda")
|
|
|
|
| 59 |
# Model
|
| 60 |
disable_torch_init()
|
| 61 |
|
| 62 |
+
model_name = get_model_name_from_path(args.model_path)
|
|
|
|
| 63 |
tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, args.model_base, model_name, args.load_8bit, args.load_4bit, device=args.device)
|
| 64 |
vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
|
| 65 |
pipe = SemanticStableDiffusionImg2ImgPipeline_DPMSolver.from_pretrained(args.sd_model_id,vae=vae,torch_dtype=torch.float16, safety_checker=None, requires_safety_checker=False).to("cuda")
|