Commit
·
db47f72
1
Parent(s):
3929e60
Update README.md
Browse files
README.md
CHANGED
@@ -14,7 +14,7 @@ import torch
|
|
14 |
from torchmetrics.functional.multimodal import clip_score
|
15 |
from functools import partial
|
16 |
|
17 |
-
model_ckpt = "raghav-gaggar/stable-diffusion-thumbs-
|
18 |
sd_pipeline = StableDiffusionPipeline.from_pretrained(model_ckpt, torch_dtype=torch.float16).to("cuda")
|
19 |
|
20 |
prompts = [
|
|
|
14 |
from torchmetrics.functional.multimodal import clip_score
|
15 |
from functools import partial
|
16 |
|
17 |
+
model_ckpt = "raghav-gaggar/stable-diffusion-thumbs-up"
|
18 |
sd_pipeline = StableDiffusionPipeline.from_pretrained(model_ckpt, torch_dtype=torch.float16).to("cuda")
|
19 |
|
20 |
prompts = [
|