BiliSakura commited on
Commit
3034d60
·
verified ·
1 Parent(s): e93cea8

Update all files for BitDance-14B-64x-diffusers

Browse files
Files changed (1) hide show
  1. bitdance_diffusers/convert.py +1 -1
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, trust_remote_code=True).to("cuda")',
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
  ]