raghav-gaggar commited on
Commit
db47f72
·
1 Parent(s): 3929e60

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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-up1"
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 = [