Skip to content

Commit

Permalink
Update convertModels.py
Browse files Browse the repository at this point in the history
  • Loading branch information
rohitgandikota authored Mar 12, 2023
1 parent 88cd155 commit 4100f4e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions train-scripts/convertModels.py
Original file line number Diff line number Diff line change
@@ -785,8 +785,8 @@ def convert_open_clip_checkpoint(checkpoint):
def savemodelDiffusers(name, compvis_config_file, diffusers_config_file, device='cpu'):
checkpoint_path = f'models/{name}/{name}.pt'

original_config_file = compvis_config_file#'/share/u/rohit/stable-diffusion/configs/stable-diffusion/v1-inference.yaml'
config_file = diffusers_config_file#'/share/u/rohit/stable-diffusion/diffusers_unet_config.json'
original_config_file = compvis_config_file
config_file = diffusers_config_file
num_in_channels = 4
scheduler_type = 'ddim'
pipeline_type = None

0 comments on commit 4100f4e

Please sign in to comment.