aljaz/generative-models
1
0
Fork 0
mirror of https://github.com/Stability-AI/generative-models.git synced 2026-02-02 04:14:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

simplify device_manager usage

Browse Source
This commit is contained in:
Stephan Auerhahn
2023-08-10 13:05:30 -07:00
parent 88395261d8
commit 3816aaa639
2 changed files with 19 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/demo/streamlit_helpers.py
View File

@@ -36,7 +36,7 @@ def init_st(
pipeline = SamplingPipeline(
model_spec=spec,
use_fp16=True,
device_manager=CudaModelManager(device="cuda", swap_device="cpu"),
device=CudaModelManager(device="cuda", swap_device="cpu"),
)
else:
pipeline = SamplingPipeline(model_spec=spec, use_fp16=False)