cocktailpeanut commited on
Commit
b535777
·
1 Parent(s): 6322f82
Files changed (1) hide show
  1. pipeline_t2i_adapter.py +3 -3
pipeline_t2i_adapter.py CHANGED
@@ -207,7 +207,7 @@ class PhotoMakerStableDiffusionXLAdapterPipeline(StableDiffusionXLAdapterPipelin
207
  # Load the main state dict first.
208
  cache_dir = kwargs.pop("cache_dir", None)
209
  force_download = kwargs.pop("force_download", False)
210
- resume_download = kwargs.pop("resume_download", False)
211
  proxies = kwargs.pop("proxies", None)
212
  local_files_only = kwargs.pop("local_files_only", None)
213
  token = kwargs.pop("token", None)
@@ -224,7 +224,7 @@ class PhotoMakerStableDiffusionXLAdapterPipeline(StableDiffusionXLAdapterPipelin
224
  weights_name=weight_name,
225
  cache_dir=cache_dir,
226
  force_download=force_download,
227
- resume_download=resume_download,
228
  proxies=proxies,
229
  local_files_only=local_files_only,
230
  token=token,
@@ -912,4 +912,4 @@ class PhotoMakerStableDiffusionXLAdapterPipeline(StableDiffusionXLAdapterPipelin
912
  if not return_dict:
913
  return (image,)
914
 
915
- return StableDiffusionXLPipelineOutput(images=image)
 
207
  # Load the main state dict first.
208
  cache_dir = kwargs.pop("cache_dir", None)
209
  force_download = kwargs.pop("force_download", False)
210
+ # resume_download = kwargs.pop("resume_download", False)
211
  proxies = kwargs.pop("proxies", None)
212
  local_files_only = kwargs.pop("local_files_only", None)
213
  token = kwargs.pop("token", None)
 
224
  weights_name=weight_name,
225
  cache_dir=cache_dir,
226
  force_download=force_download,
227
+ # resume_download=resume_download,
228
  proxies=proxies,
229
  local_files_only=local_files_only,
230
  token=token,
 
912
  if not return_dict:
913
  return (image,)
914
 
915
+ return StableDiffusionXLPipelineOutput(images=image)