codemichaeld commited on
Commit
54239e1
·
verified ·
1 Parent(s): bd8eca9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def convert_unet_to_gguf(safetensors_path, output_dir, progress=gr.Progress()):
25
 
26
  # Copy input to working dir because ggc expects files in current dir
27
  work_dir = tempfile.mkdtemp()
28
- input_path = os.path.join(work_dir, "unet.safetensors")
29
  shutil.copy(safetensors_path, input_path)
30
 
31
  # GGUF output will be named automatically like unet.safetensors -> unet.gguf
 
25
 
26
  # Copy input to working dir because ggc expects files in current dir
27
  work_dir = tempfile.mkdtemp()
28
+ input_path = os.path.join(work_dir, "unet_diffusion_pytorch_model.safetensors")
29
  shutil.copy(safetensors_path, input_path)
30
 
31
  # GGUF output will be named automatically like unet.safetensors -> unet.gguf