Skip to content

Commit

Permalink
fix paddlemix ce (#3013)
Browse files Browse the repository at this point in the history
* fix paddlemix ce

* fix paddlemix ce
  • Loading branch information
xiegegege authored Jan 8, 2025
1 parent 5ab64d8 commit cceb324
Show file tree
Hide file tree
Showing 23 changed files with 9 additions and 5 deletions.
2 changes: 1 addition & 1 deletion models/PaddleMIX/CE/ppdiffusers/cogvideo/start.sh
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ exit_code=0
echo "*******CogVideo infer begin***********"
(python infer.py \
--prompt "a bear is walking in a zoon" \
--model_path THUDM/CogVideoX-2b/ \
--model_path THUDM/CogVideoX-2b \
--generate_type "t2v" \
--dtype "float16" \
--seed 42) 2>&1 | tee ${log_dir}/CogVideo_infer.log
Expand Down
2 changes: 1 addition & 1 deletion models/PaddleMIX/CE/ppdiffusers/controlnet/infer.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
pipe = StableDiffusionControlNetPipeline.from_pretrained(
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, safety_checker=None)
canny_edged_image = load_image(
"221844474-fd539851-7649-470e-bded-4d174271cc7f.png")
"https://user-images.githubusercontent.com/50394665/221844474-fd539851-7649-470e-bded-4d174271cc7f.png")
img = pipe(prompt="pale golden rod circle with old lace background",
image=canny_edged_image, guidance_scale=9, num_inference_steps=50).images[0]
img.save("demo.png")
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
from ppdiffusers import LCMScheduler

lora_state_dict = './checkpoints/pytorch_lora_weights.safetensors'

base_model_path = 'wangqixun/YamerMIX_v8'
pipe.scheduler=LCMScheduler.from_pretrained(base_model_path,
subfolder="scheduler",
from_hf_hub=True,
Expand All @@ -18,4 +18,4 @@
pipe.fuse_lora()

num_inference_steps = 10
guidance_scale = 0
guidance_scale = 0
4 changes: 4 additions & 0 deletions models/PaddleMIX/CE/ppdiffusers/fire.sh
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,10 @@ for subdir in */; do
continue
fi

if [ "$subdir" == "deleted/" ]; then
continue
fi

# 32g显存跑不了
# if [ "$subdir" == "kandinsky2_2_text_to_image/" ]; then
# continue
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
cur_path=$(pwd)
echo ${cur_path}

work_path=${root_path}/PaddleMIX/ppdiffusers/examples/video_tokenizer/magvit2/
work_path=${root_path}/PaddleMIX/ppdiffusers/examples/visual_tokenizer/magvit2/
echo ${work_path}

log_dir=${root_path}/log
Expand Down

0 comments on commit cceb324

Please sign in to comment.