lengyue233 commited on
Commit
f2709ef
1 Parent(s): 0c92d16

reset hydra

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -306,6 +306,7 @@ if __name__ == "__main__":
306
  args.vqgan_config_name = "vqgan_pretrain"
307
 
308
  logger.info("Loading Llama model...")
 
309
  llama_queue = launch_thread_safe_queue(
310
  config_name=args.llama_config_name,
311
  checkpoint_path=args.llama_checkpoint_path,
@@ -317,6 +318,7 @@ if __name__ == "__main__":
317
  llama_tokenizer = AutoTokenizer.from_pretrained(args.tokenizer)
318
  logger.info("Llama model loaded, loading VQ-GAN model...")
319
 
 
320
  vqgan_model = load_vqgan_model(
321
  config_name=args.vqgan_config_name,
322
  checkpoint_path=args.vqgan_checkpoint_path,
 
306
  args.vqgan_config_name = "vqgan_pretrain"
307
 
308
  logger.info("Loading Llama model...")
309
+ hydra.core.global_hydra.GlobalHydra.instance().clear()
310
  llama_queue = launch_thread_safe_queue(
311
  config_name=args.llama_config_name,
312
  checkpoint_path=args.llama_checkpoint_path,
 
318
  llama_tokenizer = AutoTokenizer.from_pretrained(args.tokenizer)
319
  logger.info("Llama model loaded, loading VQ-GAN model...")
320
 
321
+ hydra.core.global_hydra.GlobalHydra.instance().clear()
322
  vqgan_model = load_vqgan_model(
323
  config_name=args.vqgan_config_name,
324
  checkpoint_path=args.vqgan_checkpoint_path,