Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Overbill1683
Stable Diffusion Webui
提交
30194529
S
Stable Diffusion Webui
项目概览
Overbill1683
/
Stable Diffusion Webui
12 个月 前同步成功
通知
1771
Star
81
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
分析
仓库
DevOps
项目成员
Pages
S
Stable Diffusion Webui
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Pages
分析
分析
仓库分析
DevOps
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
提交
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
30194529
编写于
10月 29, 2022
作者:
A
AUTOMATIC1111
提交者:
GitHub
10月 29, 2022
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #3803 from FlameLaw/master
Fixed proper dataset shuffling
上级
86e19fe8
a0a7024c
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
2 addition
and
2 deletion
+2
-2
modules/textual_inversion/dataset.py
modules/textual_inversion/dataset.py
+2
-2
未找到文件。
modules/textual_inversion/dataset.py
浏览文件 @
30194529
...
...
@@ -86,12 +86,12 @@ class PersonalizedBase(Dataset):
assert
len
(
self
.
dataset
)
>
0
,
"No images have been found in the dataset."
self
.
length
=
len
(
self
.
dataset
)
*
repeats
//
batch_size
self
.
initial_indexes
=
np
.
arange
(
len
(
self
.
dataset
)
)
self
.
dataset_length
=
len
(
self
.
dataset
)
self
.
indexes
=
None
self
.
shuffle
()
def
shuffle
(
self
):
self
.
indexes
=
self
.
initial_indexes
[
torch
.
randperm
(
self
.
initial_indexes
.
shape
[
0
]).
numpy
()]
self
.
indexes
=
np
.
random
.
permutation
(
self
.
dataset_length
)
def
create_text
(
self
,
filename_text
):
text
=
random
.
choice
(
self
.
lines
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录