Update app.py
Browse files
app.py
CHANGED
|
@@ -17,6 +17,7 @@ from cheetah.conversation.conversation_llama2 import Chat, CONV_VISION
|
|
| 17 |
from cheetah.models import *
|
| 18 |
from cheetah.processors import *
|
| 19 |
|
|
|
|
| 20 |
def parse_args():
|
| 21 |
parser = argparse.ArgumentParser(description="Demo")
|
| 22 |
parser.add_argument("--cfg-path", required=True, help="path to configuration file.")
|
|
@@ -38,6 +39,7 @@ def setup_seeds(seed = 50):
|
|
| 38 |
torch.cuda.manual_seed_all(seed)
|
| 39 |
cudnn.benchmark = False
|
| 40 |
cudnn.deterministic = True
|
|
|
|
| 41 |
|
| 42 |
print('Initializing Chat')
|
| 43 |
#args = parse_args()
|
|
|
|
| 17 |
from cheetah.models import *
|
| 18 |
from cheetah.processors import *
|
| 19 |
|
| 20 |
+
'''
|
| 21 |
def parse_args():
|
| 22 |
parser = argparse.ArgumentParser(description="Demo")
|
| 23 |
parser.add_argument("--cfg-path", required=True, help="path to configuration file.")
|
|
|
|
| 39 |
torch.cuda.manual_seed_all(seed)
|
| 40 |
cudnn.benchmark = False
|
| 41 |
cudnn.deterministic = True
|
| 42 |
+
'''
|
| 43 |
|
| 44 |
print('Initializing Chat')
|
| 45 |
#args = parse_args()
|