highperfocused/nostrdvm
1
0
Fork 0
mirror of https://github.com/believethehype/nostrdvm.git synced 2025-07-20 07:54:49 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update stablediffusionxl.py

Browse Source
This commit is contained in:
Believethehype
2023-12-08 09:47:47 +01:00
parent 65c8d1597e
commit 893c7e9e45
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backends/nserver/modules/stablediffusionxl/stablediffusionxl.py
View File

@ -43,7 +43,7 @@ class StableDiffusionXL(Processor):
def process_data(self, ds_iter) -> dict: def process_data(self, ds_iter) -> dict:
self.device = "cuda" if torch.cuda.is_available() else "cpu" self.device = "cuda" if torch.cuda.is_available() else "cpu"
self.variant = "fp16" self.variant = "fp16"
self.torch_d_type = torch.float32 self.torch_d_type = torch.float16
self.ds_iter = ds_iter self.ds_iter = ds_iter
current_session_name = self.ds_iter.session_names[0] current_session_name = self.ds_iter.session_names[0]
self.current_session = self.ds_iter.sessions[current_session_name]['manager'] self.current_session = self.ds_iter.sessions[current_session_name]['manager']