kjerk commited on
Commit
ce883f8
1 Parent(s): 97c4ba3

Small doc fix. Add more file extensions for safetensors convert.

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -22,7 +22,7 @@ def completed_download_callback():
22
  ui_filedownload_ckpt.empty()
23
 
24
  with col1:
25
- # A tool for rescaling the strength of Lora weights
26
  streamlit.html("<h3>Rescale Lora Strength</h3>")
27
  ui_fileupload_rescale = streamlit.file_uploader("Upload a safetensors lora", key="fileupload_rescale", type=[".safetensors"]) # type: UploadedFile
28
  new_scale_factor = streamlit.number_input("Scale Factor", value=1.0, step=0.01, max_value=100.0, min_value=0.01)
@@ -31,7 +31,7 @@ with col1:
31
  ui_filedownload_rescale = streamlit.empty()
32
 
33
  with col2:
34
- # A tool for removing CLIP parameters from a Lora file
35
  streamlit.html("<h3>Remove CLIP Parameters</h3>")
36
  ui_fileupload_stripclip = streamlit.file_uploader("Upload a safetensors lora", key="fileupload_stripclip", type=[".safetensors"]) # type: UploadedFile
37
 
@@ -40,9 +40,9 @@ with col2:
40
 
41
  streamlit.html("<hr>")
42
 
43
- # A tool for converting a .ckpt file to a .safetensors file
44
  streamlit.html("<h3>Convert CKPT to Safetensors (700MB max)</h3>")
45
- ui_fileupload_ckpt = streamlit.file_uploader("Upload a .ckpt file", key="fileupload_convertckpt", type=[".ckpt"]) # type: UploadedFile
46
 
47
  # Preallocate download button
48
  ui_filedownload_ckpt = streamlit.empty()
 
22
  ui_filedownload_ckpt.empty()
23
 
24
  with col1:
25
+ # - A tool for rescaling the strength of Lora weights
26
  streamlit.html("<h3>Rescale Lora Strength</h3>")
27
  ui_fileupload_rescale = streamlit.file_uploader("Upload a safetensors lora", key="fileupload_rescale", type=[".safetensors"]) # type: UploadedFile
28
  new_scale_factor = streamlit.number_input("Scale Factor", value=1.0, step=0.01, max_value=100.0, min_value=0.01)
 
31
  ui_filedownload_rescale = streamlit.empty()
32
 
33
  with col2:
34
+ # - A tool for removing CLIP parameters from a Lora file
35
  streamlit.html("<h3>Remove CLIP Parameters</h3>")
36
  ui_fileupload_stripclip = streamlit.file_uploader("Upload a safetensors lora", key="fileupload_stripclip", type=[".safetensors"]) # type: UploadedFile
37
 
 
40
 
41
  streamlit.html("<hr>")
42
 
43
+ # - A tool for converting a .ckpt file to a .safetensors file
44
  streamlit.html("<h3>Convert CKPT to Safetensors (700MB max)</h3>")
45
+ ui_fileupload_ckpt = streamlit.file_uploader("Upload a .ckpt file", key="fileupload_convertckpt", type=[".ckpt", ".pt", ".pth"]) # type: UploadedFile
46
 
47
  # Preallocate download button
48
  ui_filedownload_ckpt = streamlit.empty()