JackAILab commited on
Commit
2a25d35
1 Parent(s): 7bdb7e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -15,12 +15,14 @@ print(zero.device) # <-- 'cpu' 🤔
15
  # Gets the absolute path of the current script
16
  script_directory = os.path.dirname(os.path.realpath(__file__))
17
 
 
 
 
18
  @spaces.GPU
19
  def process(inputImage,prompt,negative_prompt):
20
 
21
  device = zero.device # "cuda"
22
  base_model_path = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
23
- consistentID_path = "JackAILab/ConsistentID/ConsistentID-v1.bin"
24
 
25
  ### Load base model
26
  pipe = ConsistentIDStableDiffusionPipeline.from_pretrained(
 
15
  # Gets the absolute path of the current script
16
  script_directory = os.path.dirname(os.path.realpath(__file__))
17
 
18
+ # download ConsistentID checkpoint to cache
19
+ consistentID_path = hf_hub_download(repo_id="JackAILab/ConsistentID", filename="ConsistentID-v1.bin", repo_type="model")
20
+
21
  @spaces.GPU
22
  def process(inputImage,prompt,negative_prompt):
23
 
24
  device = zero.device # "cuda"
25
  base_model_path = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
 
26
 
27
  ### Load base model
28
  pipe = ConsistentIDStableDiffusionPipeline.from_pretrained(