Commit
•
b859dac
1
Parent(s):
3011dc0
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -52,7 +52,7 @@ class MARModel(DiffusionPipeline):
|
|
| 52 |
diffloss_w = 1536
|
| 53 |
else:
|
| 54 |
raise NotImplementedError
|
| 55 |
-
download and load the model weights (.safetensors or .pth)
|
| 56 |
model_checkpoint_path = hf_hub_download(
|
| 57 |
repo_id=kwargs.get("repo_id", "jadechoghari/mar"),
|
| 58 |
filename=kwargs.get("model_filename", "checkpoint-last.pth")
|
|
|
|
| 52 |
diffloss_w = 1536
|
| 53 |
else:
|
| 54 |
raise NotImplementedError
|
| 55 |
+
# download and load the model weights (.safetensors or .pth)
|
| 56 |
model_checkpoint_path = hf_hub_download(
|
| 57 |
repo_id=kwargs.get("repo_id", "jadechoghari/mar"),
|
| 58 |
filename=kwargs.get("model_filename", "checkpoint-last.pth")
|