Hugging Face's logo

Spaces:
Maikou
/
Michelangelo
Running on T4

App Files Community
1
Maikou commited on
Commit
9fc50d2
•
1 Parent(s): fc1e245

path

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -33,7 +33,7 @@ image_model_config_dict = OrderedDict({
33
  # "config": "./configs/image_cond_diffuser_asl/image-ASLDM-256.yaml",
34
  # "ckpt_path": "./checkpoints/image_cond_diffuser_asl/image-ASLDM-256.ckpt",
35
  "config": "/home/user/app/configs/image_cond_diffuser_asl/image-ASLDM-256.yaml",
36
- "ckpt_path": "image_cond_diffuser_asl/image-ASLDM-256.ckpt",
37
  },
38
  })
39
 
@@ -42,11 +42,11 @@ text_model_config_dict = OrderedDict({
42
  # "config": "./configs/text_cond_diffuser_asl/text-ASLDM-256.yaml",
43
  # "ckpt_path": "./checkpoints/text_cond_diffuser_asl/text-ASLDM-256.ckpt",
44
  "config": "./configs/text_cond_diffuser_asl/text-ASLDM-256.yaml",
45
- "ckpt_path": "text_cond_diffuser_asl/text-ASLDM-256.ckpt",
46
  },
47
  })
48
 
49
- model_path = snapshot_download(repo_id="Maikou/Michelangelo/tree/main/checkpoints")
50
 
51
  class InferenceModel(object):
52
  model = None
 
33
  # "config": "./configs/image_cond_diffuser_asl/image-ASLDM-256.yaml",
34
  # "ckpt_path": "./checkpoints/image_cond_diffuser_asl/image-ASLDM-256.ckpt",
35
  "config": "/home/user/app/configs/image_cond_diffuser_asl/image-ASLDM-256.yaml",
36
+ "ckpt_path": "checkpoints/image_cond_diffuser_asl/image-ASLDM-256.ckpt",
37
  },
38
  })
39
 
 
42
  # "config": "./configs/text_cond_diffuser_asl/text-ASLDM-256.yaml",
43
  # "ckpt_path": "./checkpoints/text_cond_diffuser_asl/text-ASLDM-256.ckpt",
44
  "config": "./configs/text_cond_diffuser_asl/text-ASLDM-256.yaml",
45
+ "ckpt_path": "checkpoints/text_cond_diffuser_asl/text-ASLDM-256.ckpt",
46
  },
47
  })
48
 
49
+ model_path = snapshot_download(repo_id="Maikou/Michelangelo")
50
 
51
  class InferenceModel(object):
52
  model = None