Spaces:

chansung
/

LLaMA-7B

Runtime error

chansung commited on Mar 3, 2023

Commit

d8a82cd

1 Parent(s): 2ef1025

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ os.environ["WORKD_SIZE"] = "1"
 os.environ["MASTER_ADDR"] = "127.0.0.1"
 os.environ["MASTER_PORT"] = "50505"
-# local_rank, world_size = setup_model_parallel()
 # generator = get_pretrained_models("7B", "tokenizer")
 def chat(user_input):

 os.environ["MASTER_ADDR"] = "127.0.0.1"
 os.environ["MASTER_PORT"] = "50505"
+local_rank, world_size = setup_model_parallel()
 # generator = get_pretrained_models("7B", "tokenizer")
 def chat(user_input):