Update all files for BitDance-14B-64x-diffusers
Browse files
bitdance_diffusers/convert.py
CHANGED
|
@@ -48,7 +48,7 @@ def _copy_runtime_source(output_path: Path) -> None:
|
|
| 48 |
"",
|
| 49 |
"model_dir = Path(__file__).resolve().parent",
|
| 50 |
"sys.path.insert(0, str(model_dir))",
|
| 51 |
-
'pipe = DiffusionPipeline.from_pretrained(model_dir,
|
| 52 |
'images = pipe(prompt="A scenic mountain lake at sunrise.").images',
|
| 53 |
'images[0].save("sample.png")',
|
| 54 |
]
|
|
|
|
| 48 |
"",
|
| 49 |
"model_dir = Path(__file__).resolve().parent",
|
| 50 |
"sys.path.insert(0, str(model_dir))",
|
| 51 |
+
'pipe = DiffusionPipeline.from_pretrained(model_dir, custom_pipeline=model_dir).to("cuda")',
|
| 52 |
'images = pipe(prompt="A scenic mountain lake at sunrise.").images',
|
| 53 |
'images[0].save("sample.png")',
|
| 54 |
]
|