Spaces:
Sleeping
Sleeping
lengyue233
commited on
Commit
•
f2709ef
1
Parent(s):
0c92d16
reset hydra
Browse files
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,
|