lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-08 17:09:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

implement args for new backend

Browse Source
This commit is contained in:
layerdiffusion
2024-07-29 11:22:20 -06:00
parent 1b2610db3e
commit 5d291c28ad
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
backend/args.py Normal file
View File

@@ -0,0 +1,7 @@
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--cuda-stream", action="store_true")
args = parser.parse_known_args()[0]

7
backend/stream.py
View File

@@ -1,5 +1,5 @@
import torch
import argparse
from backend import args
def stream_context():
@@ -56,11 +56,6 @@ current_stream = None
mover_stream = None
using_stream = False
parser = argparse.ArgumentParser()
parser.add_argument("--cuda-stream", action="store_true")
args = parser.parse_known_args()[0]
if args.cuda_stream:
current_stream = get_current_stream()
mover_stream = get_new_stream()