Update README.md
Browse files
README.md
CHANGED
|
@@ -15,7 +15,7 @@ Before you begin, ensure you have downloaded the pre-trained weights for Stable
|
|
| 15 |
### 2. Prepare LoRA Weights
|
| 16 |
We have trained LoRA weights for the Patterns-1K dataset. You can download the trained LoRA weights from the following links:
|
| 17 |
- LoRA weights after 1 epoch: [1epoch_lora.safetensors](link)
|
| 18 |
-
- LoRA weights after 100 epochs: [
|
| 19 |
|
| 20 |
### 3. Test the Model
|
| 21 |
After downloading the weights, you can use the `test_lora.py` script to test the model's performance. Follow these steps:
|
|
@@ -30,4 +30,15 @@ pip install diffusers transformers torch
|
|
| 30 |
To test the model with the LoRA weights trained for 1 epoch:
|
| 31 |
```bash
|
| 32 |
python test_lora.py
|
| 33 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
### 2. Prepare LoRA Weights
|
| 16 |
We have trained LoRA weights for the Patterns-1K dataset. You can download the trained LoRA weights from the following links:
|
| 17 |
- LoRA weights after 1 epoch: [1epoch_lora.safetensors](link)
|
| 18 |
+
- LoRA weights after 100 epochs: [100epoch_lora.safetensors](link)
|
| 19 |
|
| 20 |
### 3. Test the Model
|
| 21 |
After downloading the weights, you can use the `test_lora.py` script to test the model's performance. Follow these steps:
|
|
|
|
| 30 |
To test the model with the LoRA weights trained for 1 epoch:
|
| 31 |
```bash
|
| 32 |
python test_lora.py
|
| 33 |
+
```
|
| 34 |
+
|
| 35 |
+
#### View the Results
|
| 36 |
+
The generated images will be saved to the specified paths:
|
| 37 |
+
|
| 38 |
+
Results after 1 epoch: `1epoch_test_results.png`
|
| 39 |
+
|
| 40 |
+
Results after 100 epochs: `100epoch_test_results.png`
|
| 41 |
+
|
| 42 |
+
Here are the example results:
|
| 43 |
+
- 
|
| 44 |
+
- 
|