lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 19:21:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix CQ tests (#141)

Browse Source 
* Make test client run on cpu

* test on cpu

try fix device

try fix device

try fix device

* Use real SD1.5 model for testing

* ckpt nits

* Remove coverage calls
This commit is contained in:
Chenlei Hu
2024-02-09 21:33:05 +00:00
committed by GitHub
parent 472a510151
commit ac4a8820a5
6 changed files with 89 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ldm_patched/modules/model_management.py
View File

@@ -5,6 +5,7 @@ from ldm_patched.modules.args_parser import args
import ldm_patched.modules.utils
import torch
import sys
import os
class VRAMState(Enum):
DISABLED = 0 #No vram present: no need to move models to vram
@@ -58,7 +59,7 @@ try:
except:
pass
if args.always_cpu:
if args.always_cpu or os.environ.get("FORGE_CQ_TEST", ""):
cpu_state = CPUState.CPU
def is_intel_xpu():