Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
H
Hydra Node Http
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Locked Files
Issues
0
Issues
0
List
Boards
Labels
Service Desk
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Security & Compliance
Security & Compliance
Dependency List
License Compliance
Packages
Packages
List
Container Registry
Analytics
Analytics
CI / CD
Code Review
Insights
Issues
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
novelai-storage
Hydra Node Http
Commits
807acb90
Commit
807acb90
authored
Sep 17, 2022
by
kurumuz
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
change
parent
44328813
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
9 deletions
+7
-9
hydra_node/models.py
hydra_node/models.py
+7
-9
No files found.
hydra_node/models.py
View file @
807acb90
...
...
@@ -19,6 +19,11 @@ import k_diffusion as K
import
contextlib
import
random
def
seed_everything
(
seed
:
int
):
torch
.
manual_seed
(
seed
)
np
.
random
.
seed
(
seed
)
random
.
seed
(
seed
)
def
pil_upscale
(
image
,
scale
=
1
):
device
=
image
.
device
dtype
=
image
.
dtype
...
...
@@ -64,16 +69,14 @@ def prompt_mixing(model, prompt_body, batch_size):
def
sample_start_noise
(
seed
,
C
,
H
,
W
,
f
,
device
=
"cuda"
):
if
seed
:
torch
.
manual_seed
(
seed
)
np
.
random
.
seed
(
seed
)
seed_everything
(
seed
)
noise
=
torch
.
randn
([
C
,
(
H
)
//
f
,
(
W
)
//
f
],
device
=
device
)
.
unsqueeze
(
0
)
return
noise
def
sample_start_noise_special
(
seed
,
request
,
device
=
"cuda"
):
if
seed
:
torch
.
manual_seed
(
seed
)
np
.
random
.
seed
(
seed
)
seed_everything
(
seed
)
noise
=
torch
.
randn
([
request
.
latent_channels
,
request
.
height
//
request
.
downsampling_factor
,
request
.
width
//
request
.
downsampling_factor
],
device
=
device
)
.
unsqueeze
(
0
)
return
noise
...
...
@@ -163,11 +166,6 @@ class StableInterface(nn.Module):
return
x_0
def
seed_everything
(
seed
:
int
):
torch
.
manual_seed
(
seed
)
np
.
random
.
seed
(
seed
)
random
.
seed
(
seed
)
class
StableDiffusionModel
(
nn
.
Module
):
def
__init__
(
self
,
config
):
nn
.
Module
.
__init__
(
self
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment