storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-17 00:23:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

add debugging to sd python script

Browse Source
This commit is contained in:
Brandon Thomas
2024-02-10 02:23:50 -05:00
parent 87591b6991
commit f6e9c1ce5c
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
image_generation/Storyteller-Studio-Filter/main.py
+15
View File
@@ -2,12 +2,27 @@ import argparse
import os
import pathlib
import torch
import sys
# Note please use type hints.
from stablediffusionfe import StableDiffusionFE
import random
import string
def print_gpu_info():
print('========================================')
print('Python Interpreter', sys.executable)
print('Python Version', sys.version)
print('Python Version Info', sys.version_info)
print('PyTorch Version', torch.__version__)
print('CUDA Available?', torch.cuda.is_available())
print('CUDA Version', torch.version.cuda)
print('CUDA Device Count', torch.cuda.device_count())
print('CUDA architectures library was compiled for', torch.cuda.get_arch_list())
print('========================================', flush=True)
print_gpu_info()
def generate_random_string(length=6):
# Choose from uppercase, lowercase letters and digits
characters = string.ascii_letters + string.digits