remi349 commited on
Commit
bf3bb91
·
verified ·
1 Parent(s): 8e749a0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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('remi349/sd_trained_3D_lora')
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 with [Josué ADOSSEHOUN](https://huggingface.co/josh007)
 
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)