Nef Caballero commited on
Commit
178149f
·
1 Parent(s): 62aa27a

fix attempt for checkpoint loading 2

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ from huggingface_hub import hf_hub_download
8
  import spaces
9
  from comfy import model_management
10
 
11
- hf_hub_download(repo_id="Comfy-Org/stable-diffusion-v1-5-archive", filename="v1-5-pruned.safetensors", local_dir="models/checkpoints")
12
 
13
 
14
  def get_value_at_index(obj: Union[Sequence, Mapping], index: int) -> Any:
@@ -122,7 +122,7 @@ from nodes import NODE_CLASS_MAPPINGS
122
 
123
  checkpointloadersimple = NODE_CLASS_MAPPINGS["CheckpointLoaderSimple"]()
124
  checkpointloadersimple_4 = checkpointloadersimple.load_checkpoint(
125
- ckpt_name="v1-5-pruned.safetensors"
126
  )
127
 
128
 
 
8
  import spaces
9
  from comfy import model_management
10
 
11
+ hf_hub_download(repo_id="Comfy-Org/stable-diffusion-v1-5-archive", filename="v1-5-pruned-emaonly-fp16.safetensors", local_dir="models/checkpoints")
12
 
13
 
14
  def get_value_at_index(obj: Union[Sequence, Mapping], index: int) -> Any:
 
122
 
123
  checkpointloadersimple = NODE_CLASS_MAPPINGS["CheckpointLoaderSimple"]()
124
  checkpointloadersimple_4 = checkpointloadersimple.load_checkpoint(
125
+ ckpt_name="v1-5-pruned-emaonly-fp16.safetensors"
126
  )
127
 
128