recoilme
commited on
Commit
·
674f21a
1
Parent(s):
2b35a28
3009
Browse files- samples/unet_192x384_0.jpg +2 -2
- samples/unet_256x384_0.jpg +2 -2
- samples/unet_320x384_0.jpg +2 -2
- samples/unet_384x192_0.jpg +2 -2
- samples/unet_384x256_0.jpg +2 -2
- samples/unet_384x320_0.jpg +2 -2
- samples/unet_384x384_0.jpg +2 -2
- src/dataset_from_folder.py +4 -4
- unet/diffusion_pytorch_model.safetensors +1 -1
samples/unet_192x384_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_256x384_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_320x384_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_384x192_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_384x256_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_384x320_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_384x384_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
src/dataset_from_folder.py
CHANGED
|
@@ -21,14 +21,14 @@ from datetime import timedelta
|
|
| 21 |
dtype = torch.float16
|
| 22 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 23 |
batch_size = 5
|
| 24 |
-
min_size = 192 #256 #192
|
| 25 |
-
max_size = 384 #256 #384
|
| 26 |
-
step = 64
|
| 27 |
empty_share = 0.05
|
| 28 |
limit = 0
|
| 29 |
# Основная процедура обработки
|
| 30 |
folder_path = "/workspace/animesfw" #alchemist"
|
| 31 |
-
save_path = "/workspace/
|
| 32 |
os.makedirs(save_path, exist_ok=True)
|
| 33 |
|
| 34 |
# Функция для очистки CUDA памяти
|
|
|
|
| 21 |
dtype = torch.float16
|
| 22 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 23 |
batch_size = 5
|
| 24 |
+
min_size = 384 #192 #256 #192
|
| 25 |
+
max_size = 640 #384 #256 #384
|
| 26 |
+
step = 128 #64
|
| 27 |
empty_share = 0.05
|
| 28 |
limit = 0
|
| 29 |
# Основная процедура обработки
|
| 30 |
folder_path = "/workspace/animesfw" #alchemist"
|
| 31 |
+
save_path = "/workspace/animesfw640" #"alchemist"
|
| 32 |
os.makedirs(save_path, exist_ok=True)
|
| 33 |
|
| 34 |
# Функция для очистки CUDA памяти
|
unet/diffusion_pytorch_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 6184944280
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9a08ef781fc71bca62533dd8987750d3c57697b4a051add2449410872f97b110
|
| 3 |
size 6184944280
|