FilesExpand file tree

 main

/

test_system_info.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

26 lines (20 loc) · 763 Bytes

 main

/

test_system_info.py

Top

File metadata and controls

• Code
• Blame

26 lines (20 loc) · 763 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

import os

import traceback

import stable_diffusion_cpp.stable_diffusion_cpp as sd_cpp

try:

# Get system info

system_info = sd_cpp.sd_get_system_info()

num_physical_cores = sd_cpp.get_num_physical_cores()

# Print system info

print("System info: ", system_info)

print("Number of physical cores: ", num_physical_cores)

OUTPUT_DIR = "tests/outputs"

if not os.path.exists(OUTPUT_DIR):

os.makedirs(OUTPUT_DIR)

# Write system info to file txt

with open(f"{OUTPUT_DIR}/system_info.txt", "w") as f:

f.write(f"System info: {str(system_info)}\n")

f.write(f"Number of physical cores: {str(num_physical_cores)}")

except Exception as e:

traceback.print_exc()

print("Test - system_info failed: ", e)