RiddleHe commited on
Commit
bc1e1cb
·
verified ·
1 Parent(s): e3db5b9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -34,10 +34,10 @@ These are LoRA adaption weights for CompVis/stable-diffusion-v1-4. The weights w
34
 
35
  ```python
36
  pipe = DiffusionPipeline.from_pretrained(
37
- CompVis/stable-diffusion-v1-4, torch_dtype=torch.float16
38
  )
39
 
40
- pipe.load_lora_weights(args.lora_model_name_or_path)
41
  pipe.to('cuda')
42
 
43
  prompt = "A histopathology image of breast cancer tissue"
 
34
 
35
  ```python
36
  pipe = DiffusionPipeline.from_pretrained(
37
+ "CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16
38
  )
39
 
40
+ pipe.load_lora_weights("RiddleHe/SD14_pathology_lora")
41
  pipe.to('cuda')
42
 
43
  prompt = "A histopathology image of breast cancer tissue"