Update README.md

README.md

@@ -38,7 +38,7 @@ from diffusers import StableDiffusionPipeline
 pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16)
 
 # Then add the lora weights to the model stable diffusion 2
-pipe.unet.load_attn_procs('ACROSS-Lab/sd_finetuned')
+pipe.unet.load_attn_procs('ACROSS-Lab/PromptTo3D_sd_finetuned')
 pipe.to("cuda")
 
 # Then you can begin the inference process on a prompt and save the image generated
@@ -57,7 +57,7 @@ The data used to train this model is the dataset available on uggingface at 'rem
 you can download it thanks to the command
 ```python
 from datasets import load_dataset
-dataset = load_dataset("ACROSS_Lab/finetuning_sd_dataset_objaverse", split = 'train')
+dataset = load_dataset("ACROSS_Lab/PromptTo3D_sd_dataset", split = 'train')
 ```
 
 This dataset is a subset of the dataset [Objaverse](https://objaverse.allenai.org/).