Update README.md
Browse files
README.md
CHANGED
|
@@ -38,7 +38,7 @@ from diffusers import StableDiffusionPipeline
|
|
| 38 |
pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16)
|
| 39 |
|
| 40 |
# Then add the lora weights to the model stable diffusion 2
|
| 41 |
-
pipe.unet.load_attn_procs('
|
| 42 |
pipe.to("cuda")
|
| 43 |
|
| 44 |
# Then you can begin the inference process on a prompt and save the image generated
|
|
@@ -63,4 +63,4 @@ dataset = load_dataset("remi349/finetuning_dataset_for_3D_training", split = 'tr
|
|
| 63 |
This dataset is a subset of the dataset [Objaverse](https://objaverse.allenai.org/).
|
| 64 |
|
| 65 |
## Collaboration
|
| 66 |
-
This model and dataset has been made in collaboration
|
|
|
|
| 38 |
pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16)
|
| 39 |
|
| 40 |
# Then add the lora weights to the model stable diffusion 2
|
| 41 |
+
pipe.unet.load_attn_procs('ACROSS-Lab/sd_finetuned')
|
| 42 |
pipe.to("cuda")
|
| 43 |
|
| 44 |
# Then you can begin the inference process on a prompt and save the image generated
|
|
|
|
| 63 |
This dataset is a subset of the dataset [Objaverse](https://objaverse.allenai.org/).
|
| 64 |
|
| 65 |
## Collaboration
|
| 66 |
+
This model and dataset has been made in collaboration by [Josué ADOSSEHOUN](https://huggingface.co/josh007) and [Rémi DUCOTTET](https://huggingface.co/remi349)
|