LLaMA-13B

Runtime error

chansung commited on Mar 6, 2023

Commit

0e0dd38

1 Parent(s): b50dce8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 import gradio as gr
 from strings import TITLE, ABSTRACT
-from gen import get_pretrained_models, get_output, setup_model_parallel
 os.environ["RANK"] = "0"
 os.environ["WORLD_SIZE"] = "1"

 import gradio as gr
 from strings import TITLE, ABSTRACT
+from gen import get_pretrained_models, get_output
 os.environ["RANK"] = "0"
 os.environ["WORLD_SIZE"] = "1"