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(
|