guyyariv commited on
Commit
d833a17
1 Parent(s): 2821e52

AudioTokenDemo

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -69,12 +69,12 @@ class AudioTokenWrapper(torch.nn.Module):
69
  self.unet.set_attn_processor(lora_attn_procs)
70
  self.lora_layers = AttnProcsLayers(self.unet.attn_processors)
71
  self.lora_layers.eval()
72
- lora_layers_learned_embeds = 'models/embedder_learned_embeds.bin'
73
  self.lora_layers.load_state_dict(torch.load(lora_layers_learned_embeds, map_location=device))
74
  self.unet.load_attn_procs(lora_layers_learned_embeds)
75
 
76
  self.embedder.eval()
77
- embedder_learned_embeds = 'models/lora_layers_learned_embeds.bin'
78
  self.embedder.load_state_dict(torch.load(embedder_learned_embeds, map_location=device))
79
 
80
  self.placeholder_token = '<*>'
@@ -141,4 +141,3 @@ if __name__ == "__main__":
141
  examples=examples
142
  )
143
  demo.launch()
144
-
 
69
  self.unet.set_attn_processor(lora_attn_procs)
70
  self.lora_layers = AttnProcsLayers(self.unet.attn_processors)
71
  self.lora_layers.eval()
72
+ lora_layers_learned_embeds = 'models/lora_layers_learned_embeds.bin'
73
  self.lora_layers.load_state_dict(torch.load(lora_layers_learned_embeds, map_location=device))
74
  self.unet.load_attn_procs(lora_layers_learned_embeds)
75
 
76
  self.embedder.eval()
77
+ embedder_learned_embeds = 'models/embedder_learned_embeds.bin'
78
  self.embedder.load_state_dict(torch.load(embedder_learned_embeds, map_location=device))
79
 
80
  self.placeholder_token = '<*>'
 
141
  examples=examples
142
  )
143
  demo.launch()