Rename maybe_load_lora to maybe_use_sdxl_lora (#153)

Signed-off-by: kunjan@ucla.edu <kunjanp@google.com>

Author: coolkp

src/maxdiffusion/generate_sdxl.py

@@ -33,7 +33,7 @@
     get_add_time_ids,
     rescale_noise_cfg,
     load_sdxllightning_unet,
-    maybe_load_lora,
+    maybe_load_sdxl_lora,
     create_scheduler,
 )
 
@@ -234,7 +234,7 @@ def run(config):
     params["unet"] = unet_params
 
   # maybe load lora and create interceptor
-  params, lora_interceptors = maybe_load_lora(config, pipeline, params)
+  params, lora_interceptors = maybe_load_sdxl_lora(config, pipeline, params)
 
   if config.lightning_repo:
     pipeline, params = load_sdxllightning_unet(config, pipeline, params)

src/maxdiffusion/maxdiffusion_utils.py

@@ -39,7 +39,7 @@ def load_sdxllightning_unet(config, pipeline, params):
   return pipeline, params
 
 
-def maybe_load_lora(config, pipeline, params):
+def maybe_load_sdxl_lora(config, pipeline, params):
 
   def _noop_interceptor(next_fn, args, kwargs, context):
     return next_fn(*args, **kwargs)