Skip to content

Commit

Permalink
add more demo prompts
Browse files Browse the repository at this point in the history
  • Loading branch information
dm18 committed Dec 31, 2023
1 parent 65ff21c commit 2a34cef
Showing 1 changed file with 19 additions and 0 deletions.
19 changes: 19 additions & 0 deletions examples/WebcamImg2Img/webcam.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ def convert_from_image_to_cv2(img: Image) -> np.ndarray:
#KBlueLeaf/kohaku-v2.1
#stabilityai/sdxl-turbo
#KBlueLeaf/kohaku-v3-rev2
#thibaud/sdxl_dpo_turbo
pipe = StableDiffusionPipeline.from_pretrained("KBlueLeaf/kohaku-v3-rev2").to(
device=torch.device("cuda"),
dtype=torch.float16,
Expand Down Expand Up @@ -187,6 +188,24 @@ def convert_from_image_to_cv2(img: Image) -> np.ndarray:
prompt=prompt,
negative_prompt=negative_prompt,
)
elif (key == ord('7') ):
prompt = "frieza, golden frieza, 1boy, alien, aura, red eyes, big eyes, frown, gold skin, looking at viewer, solo"
stream.prepare(
prompt=prompt,
negative_prompt=negative_prompt,
)
elif (key == ord('8') ):
prompt = "1other, alien, monster, no humans, teeth, ribs, xenomorph, hr giger"
stream.prepare(
prompt=prompt,
negative_prompt=negative_prompt,
)
elif (key == ord('9') ):
prompt = "cat, black cat, cat ears, cat girl, cat tail, cat nose, whiskers, black fur, black body, black skin, solo, collar, big eyes, black hair, fangs, animal, animal ears, 1girl"
stream.prepare(
prompt=prompt,
negative_prompt=negative_prompt,
)


# After the loop release the cap object
Expand Down

0 comments on commit 2a34cef

Please sign in to comment.