treasuraid commited on
Commit
6631c27
·
1 Parent(s): 3be7215

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ LoRa_DIR = 'VESSL/lora-fintuning-bayc'
15
  DATASET_REPO = 'VESSL/Bored_Ape_NFT_text'
16
  SAMPLE_IMAGE = 'weights/Sample.png'
17
 
18
- @st.cache(hash_funcs={torch.nn.parameter.Parameter: lambda parameter: parameter.data.numpy()})
19
  def load_pipeline_w_lora() :
20
 
21
  # Load pipeline
 
15
  DATASET_REPO = 'VESSL/Bored_Ape_NFT_text'
16
  SAMPLE_IMAGE = 'weights/Sample.png'
17
 
18
+ # @st.cache(hash_funcs={torch.nn.parameter.Parameter: lambda parameter: parameter.data.numpy()})
19
  def load_pipeline_w_lora() :
20
 
21
  # Load pipeline