Update README.md

README.md

@@ -55,33 +55,28 @@ Advanced Version: Upgraded from the base version of Imagine AI Art Generator
 ### Diffusers
 
 ```
-pip install diffusers transformers accelerate --upgrade
+pip install -r requirements.txt --no-cache
 ```
 
 - **Text-to-image**:
 
-To run the model yourself, you can leverage the 🧨 Diffusers library:
-1. Install the library:
-```
-pip install --upgrade diffusers  
-```
-
-2. Run the model:
+1. Run the model:
 ```py
-from TBA import CustomStableDiffusionPipeline4_1
+from stable_diffusion_custom_v4_1 import CustomStableDiffusionPipeline4_1
 import torch
 
-pipe = DiffusionPipeline.from_pretrained("vyroAI/ImagineV4.1")
+pipe = CustomStableDiffusionPipeline4_1.from_pretrained("vyroAI/ImagineV4.1")
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 # To save GPU memory, torch.float16 can be used, but it may compromise image quality.
-pipe.to(torch_device="cuda", torch_dtype=torch.float16)
+pipe.to(torch_device=DEVICE, torch_dtype=torch.float32)
 
 prompt = "Self-portrait oil painting, a beautiful cyborg with golden hair, 8k"
 
 # Can be set to 1~50 steps. 
-num_inference_steps = 30 
+num_inference_steps = 20 
 
-images = pipe(prompt=prompt, num_inference_steps=num_inference_steps).images
+res = pipe.inferPipe(prompt=prompt, num_inference_steps=num_inference_steps).images
 ```
 
 ### Out-of-Scope Use