Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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(
|