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="""
|