Update app.py
Browse files
app.py
CHANGED
|
@@ -10,6 +10,7 @@ dtype = torch.float16
|
|
| 10 |
|
| 11 |
repo = "stabilityai/stable-diffusion-3-medium-diffusers"
|
| 12 |
pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16).to(device)
|
|
|
|
| 13 |
|
| 14 |
MAX_SEED = np.iinfo(np.int32).max
|
| 15 |
MAX_IMAGE_SIZE = 1344
|
|
@@ -35,9 +36,8 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
|
|
| 35 |
return image, seed
|
| 36 |
|
| 37 |
examples = [
|
| 38 |
-
"
|
| 39 |
-
"
|
| 40 |
-
"A delicious ceviche cheesecake slice",
|
| 41 |
]
|
| 42 |
|
| 43 |
css="""
|
|
|
|
| 10 |
|
| 11 |
repo = "stabilityai/stable-diffusion-3-medium-diffusers"
|
| 12 |
pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16).to(device)
|
| 13 |
+
pipe.load_lora_weights("kaytoo2022/cara-the-cavapoo-2")
|
| 14 |
|
| 15 |
MAX_SEED = np.iinfo(np.int32).max
|
| 16 |
MAX_IMAGE_SIZE = 1344
|
|
|
|
| 36 |
return image, seed
|
| 37 |
|
| 38 |
examples = [
|
| 39 |
+
"Cara the Cavapoo in a jungle, cold color palette, muted colors, detailed, 8k",
|
| 40 |
+
"Cara the Cavapoo riding a green horse"
|
|
|
|
| 41 |
]
|
| 42 |
|
| 43 |
css="""
|