SYMBOL INDEX (157 symbols across 34 files) FILE: api/app.py function tinyVae (line 73) | def tinyVae(origVae: AutoencoderKL): function init (line 93) | def init(): function decodeBase64Image (line 134) | def decodeBase64Image(imageStr: str, name: str) -> PIL.Image: function getFromUrl (line 140) | def getFromUrl(url: str, name: str) -> PIL.Image: function truncateInputs (line 147) | def truncateInputs(inputs: dict): function inference (line 169) | async def inference(all_inputs: dict, response) -> dict: FILE: api/convert_to_diffusers.py function main (line 23) | def main( FILE: api/download.py function send (line 27) | async def send(type: str, status: str, payload: dict = {}, send_opts: di... function normalize_model_id (line 34) | def normalize_model_id(model_id: str, model_revision): function download_model (line 41) | async def download_model( FILE: api/download_checkpoint.py function main (line 8) | def main(checkpoint_url: str): FILE: api/extras/upsample/upsample.py function cache_path (line 35) | def cache_path(filename): function assert_model_exists (line 39) | async def assert_model_exists(src, filename, send_opts, opts={}): function download_models (line 49) | async def download_models(send_opts={}): function upsample (line 86) | async def upsample(model_inputs, call_inputs, send_opts={}, startRequest... FILE: api/getPipeline.py function listAvailablePipelines (line 15) | def listAvailablePipelines(): function availableCommunityPipelines (line 27) | def availableCommunityPipelines(): function clearPipelines (line 40) | def clearPipelines(): function getPipelineClass (line 51) | def getPipelineClass(pipeline_name: str): function getPipelineForModel (line 58) | def getPipelineForModel( FILE: api/getScheduler.py function initScheduler (line 39) | def initScheduler(MODEL_ID: str, scheduler_id: str, download=False): function getScheduler (line 65) | def getScheduler(MODEL_ID: str, scheduler_id: str, download=False): FILE: api/lib/prompts.py function prepare_prompts (line 4) | def prepare_prompts(pipeline, model_inputs, is_sdxl): FILE: api/lib/textual_inversions.py function strMap (line 17) | def strMap(str: str): function extract_tokens_from_list (line 24) | def extract_tokens_from_list(textual_inversions: list): function handle_textual_inversions (line 28) | async def handle_textual_inversions(textual_inversions: list, model, sta... FILE: api/lib/textual_inversions_test.py class TextualInversionsTest (line 5) | class TextualInversionsTest(unittest.TestCase): method test_extract_tokens_query_fname (line 6) | def test_extract_tokens_query_fname(self): method test_extract_tokens_query_token (line 11) | def test_extract_tokens_query_token(self): FILE: api/loadModel.py function loadModel (line 28) | def loadModel( FILE: api/precision.py function revision_from_precision (line 18) | def revision_from_precision(precision=MODEL_PRECISION): function torch_dtype_from_precision (line 23) | def torch_dtype_from_precision(precision=MODEL_PRECISION): function torch_dtype_from_precision (line 29) | def torch_dtype_from_precision(precision=MODEL_PRECISION): FILE: api/send.py function get_now (line 19) | def get_now(): function clearSession (line 50) | def clearSession(force=False): function getTimings (line 60) | def getTimings(): function send (line 74) | async def send(type: str, status: str, payload: dict = {}, opts: dict = ... FILE: api/server.py function healthcheck (line 27) | def healthcheck(request): function inference (line 39) | async def inference(request): FILE: api/status.py class Status (line 1) | class Status: method __init__ (line 2) | def __init__(self): method update (line 6) | def update(self, type, progress): method get (line 10) | def get(self): FILE: api/tests.py function test_memory_free_on_swap_model (line 4) | def test_memory_free_on_swap_model(): FILE: api/train_dreambooth.py function send (line 78) | def send(type: str, status: str, payload: dict = {}, send_opts: dict = {}): function TrainDreamBooth (line 82) | def TrainDreamBooth(model_id: str, pipeline, model_inputs, call_inputs, ... function save_model_card (line 266) | def save_model_card( function log_validation (line 304) | def log_validation( function import_model_class_from_model_name_or_path (line 408) | def import_model_class_from_model_name_or_path( class DreamBoothDataset (line 436) | class DreamBoothDataset(Dataset): method __init__ (line 442) | def __init__( method __len__ (line 502) | def __len__(self): method __getitem__ (line 505) | def __getitem__(self, index): function collate_fn (line 551) | def collate_fn(examples, with_prior_preservation=False): class PromptDataset (line 586) | class PromptDataset(Dataset): method __init__ (line 589) | def __init__(self, prompt, num_samples): method __len__ (line 593) | def __len__(self): method __getitem__ (line 596) | def __getitem__(self, index): function model_has_vae (line 603) | def model_has_vae(args): function tokenize_prompt (line 617) | def tokenize_prompt(tokenizer, prompt, tokenizer_max_length=None): function encode_prompt (line 634) | def encode_prompt( function main (line 653) | def main(args, init_pipeline, send_opts): FILE: api/utils/storage/BaseStorage.py class BaseArchive (line 8) | class BaseArchive(ABC): method __init__ (line 9) | def __init__(self, path, status=None): method updateStatus (line 13) | def updateStatus(self, type, progress): method extract (line 17) | def extract(self): method splitext (line 20) | def splitext(self): class TarArchive (line 26) | class TarArchive(BaseArchive): method test (line 28) | def test(path): method extract (line 31) | def extract(self, dir, dry_run=False): function Archive (line 63) | def Archive(path, **kwargs): class BaseStorage (line 69) | class BaseStorage(ABC): method test (line 72) | def test(url): method __init__ (line 75) | def __init__(self, url, **kwargs): method updateStatus (line 80) | def updateStatus(self, type, progress): method splitext (line 84) | def splitext(self): method get_filename (line 89) | def get_filename(self): method download_file (line 93) | def download_file(self, dest): method download_and_extract (line 97) | def download_and_extract(self, fname, dir=None, dry_run=False): FILE: api/utils/storage/BaseStorage_test.py class BaseStorageTest (line 5) | class BaseStorageTest(unittest.TestCase): method test_get_filename (line 6) | def test_get_filename(self): class Download_and_extract (line 10) | class Download_and_extract(unittest.TestCase): method test_file_only (line 11) | def test_file_only(self): method test_file_archive (line 16) | def test_file_archive(self): FILE: api/utils/storage/HTTPStorage.py function get_now (line 10) | def get_now(): class HTTPStorage (line 14) | class HTTPStorage(BaseStorage): method test (line 16) | def test(url): method __init__ (line 19) | def __init__(self, url, **kwargs): method upload_file (line 26) | def upload_file(self, source, dest): method download_file (line 29) | def download_file(self, fname): FILE: api/utils/storage/S3Storage.py function get_now (line 18) | def get_now(): class S3Storage (line 22) | class S3Storage(BaseStorage): method test (line 23) | def test(url): method __init__ (line 26) | def __init__(self, url, **kwargs): method s3resource (line 57) | def s3resource(self): method s3client (line 68) | def s3client(self): method bucket (line 79) | def bucket(self): method upload_file (line 86) | def upload_file(self, source, dest): method download_file (line 109) | def download_file(self, dest): method file_exists (line 129) | def file_exists(self): FILE: api/utils/storage/S3Storage_test.py class S3StorageTest (line 6) | class S3StorageTest(unittest.TestCase): method test_endpoint_only_s3 (line 7) | def test_endpoint_only_s3(self): method test_endpoint_only_http_s3 (line 13) | def test_endpoint_only_http_s3(self): method test_endpoint_only_https_s3 (line 19) | def test_endpoint_only_https_s3(self): method test_bucket_only (line 25) | def test_bucket_only(self): method test_url_with_bucket_and_file_only (line 31) | def test_url_with_bucket_and_file_only(self): method test_full_url_with_subdirectory (line 37) | def test_full_url_with_subdirectory(self): FILE: api/utils/storage/__init__.py function Storage (line 9) | def Storage(url, no_raise=False, **kwargs): FILE: api/utils/storage/__init__test.py class StorageTest (line 5) | class StorageTest(unittest.TestCase): method test_url_s3 (line 6) | def test_url_s3(self): method test_url_http (line 10) | def test_url_http(self): method test_no_match_raise (line 14) | def test_no_match_raise(self): method test_no_match_no_raise (line 18) | def test_no_match_no_raise(self): FILE: test.py function b64encode_file (line 28) | def b64encode_file(filename: str): function output_path (line 38) | def output_path(filename: str): function sizeof_fmt (line 43) | def sizeof_fmt(num, suffix="B"): function decode_and_save (line 51) | def decode_and_save(image_byte_string: str, name: str): function test (line 68) | def test(name, inputs): function runTest (line 73) | def runTest(name, args, extraCallInputs, extraModelInputs): function main (line 403) | def main(tests_to_run, args, extraCallInputs, extraModelInputs): FILE: tests/integration/conftest.py function my_fixture (line 7) | def my_fixture(): FILE: tests/integration/lib.py function log_streamer (line 40) | def log_streamer(container, name=None): function get_free_port (line 109) | def get_free_port(): function startContainer (line 117) | def startContainer(image, command=None, stream_logs=False, onstop=None, ... function getMinio (line 161) | def getMinio(id="disposable"): function getDDA (line 256) | def getDDA( function cleanup (line 357) | def cleanup(): FILE: tests/integration/test_attn_procs.py class TestAttnProcs (line 8) | class TestAttnProcs: method setup_class (line 9) | def setup_class(self): method teardown_class (line 21) | def teardown_class(self): method test_lora_hf_download (line 26) | def test_lora_hf_download(self): method test_lora_http_download_pytorch_bin (line 50) | def test_lora_http_download_pytorch_bin(self): method test_lora_http_download_civitai_safetensors (line 75) | def test_lora_http_download_civitai_safetensors(self): FILE: tests/integration/test_build_download.py function test_cloudcache_build_download (line 6) | def test_cloudcache_build_download(): function test_huggingface_build_download (line 51) | def test_huggingface_build_download(): function test_checkpoint_url_build_download (line 97) | def test_checkpoint_url_build_download(): FILE: tests/integration/test_cloud_cache.py function test_cloud_cache_create_and_upload (line 6) | def test_cloud_cache_create_and_upload(): FILE: tests/integration/test_dreambooth.py class TestDreamBoothS3 (line 8) | class TestDreamBoothS3: method setup_class (line 13) | def setup_class(self): method teardown_class (line 17) | def teardown_class(self): method test_training_s3 (line 21) | def test_training_s3(self): method test_s3_download_and_inference (line 48) | def test_s3_download_and_inference(self): class TestDreamBoothHF (line 73) | class TestDreamBoothHF: method test_training_hf (line 74) | def test_training_hf(self): method test_hf_download_and_inference (line 103) | def test_hf_download_and_inference(self): FILE: tests/integration/test_general.py class TestGeneralClass (line 7) | class TestGeneralClass: method setup_class (line 22) | def setup_class(self): method teardown_class (line 34) | def teardown_class(self): method test_txt2img (line 39) | def test_txt2img(self): method test_img2img (line 43) | def test_img2img(self): FILE: tests/integration/test_loras.py class TestLoRAs (line 7) | class TestLoRAs: method setup_class (line 8) | def setup_class(self): method teardown_class (line 20) | def teardown_class(self): method test_lora_hf_download (line 27) | def test_lora_hf_download(self): method test_lora_http_download_pytorch_bin (line 53) | def test_lora_http_download_pytorch_bin(self): method test_lora_http_download_civitai_safetensors (line 77) | def test_lora_http_download_civitai_safetensors(self): FILE: tests/integration/test_memory.py function test_memory (line 7) | def test_memory():