Hugging Face's logo

Spaces:
chansung
/
LLaMA-13B
Runtime error

App Files Community
chansung commited on
Commit
fde2ede
•
1 Parent(s): 0538992

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -11,8 +11,7 @@ os.environ["WORLD_SIZE"] = "1"
11
  os.environ["MASTER_ADDR"] = "127.0.0.1"
12
  os.environ["MASTER_PORT"] = "50505"
13
 
14
- local_rank, world_size = setup_model_parallel()
15
- generator = get_pretrained_models("7B", "tokenizer", local_rank, world_size)
16
 
17
  history = []
18
 
 
11
  os.environ["MASTER_ADDR"] = "127.0.0.1"
12
  os.environ["MASTER_PORT"] = "50505"
13
 
14
+ generator = get_pretrained_models("13B", "tokenizer")
 
15
 
16
  history = []
17