SivilTaram commited on
Commit
3b2cf1a
1 Parent(s): 61aefa1

update files

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -11,6 +11,7 @@ import redirect as rd
11
  import torch
12
  import shutil
13
  import os
 
14
 
15
 
16
  css = """
@@ -110,15 +111,19 @@ Infer the date from context. Q: Today is the second day of the third month of 1
110
  }
111
  st.table(df)
112
 
 
 
 
 
113
  # zip the final lora module
114
- torch.save(final_lora, "lora/adapter_model.bin")
115
  # create a zip file
116
- shutil.make_archive("lora_module", 'zip', "lora")
117
  with open("lora_module.zip", "rb") as fp:
118
  btn = st.download_button(
119
  label="📥 Download the final LoRA Module",
120
  data=fp,
121
- file_name="lora_module.zip",
122
  mime="application/zip"
123
  )
124
  st.warning("The page will be refreshed once you click the download button.")
 
11
  import torch
12
  import shutil
13
  import os
14
+ import uuid
15
 
16
 
17
  css = """
 
111
  }
112
  st.table(df)
113
 
114
+ random_id = uuid.uuid4().hex
115
+ os.makedirs(f"lora/{random_id}")
116
+ # copy config file
117
+ shutil.copyfile("lora/adapter_config.json", f"lora/{random_id}/adapter_config.json")
118
  # zip the final lora module
119
+ torch.save(final_lora, f"lora/{random_id}/adapter_model.bin")
120
  # create a zip file
121
+ shutil.make_archive(f"lora_{random_id}", 'zip', f"lora/{random_id}")
122
  with open("lora_module.zip", "rb") as fp:
123
  btn = st.download_button(
124
  label="📥 Download the final LoRA Module",
125
  data=fp,
126
+ file_name=f"lora_{random_id}.zip",
127
  mime="application/zip"
128
  )
129
  st.warning("The page will be refreshed once you click the download button.")