chansung commited on
Commit
0483827
1 Parent(s): f57923a

Update gen.py

Browse files
Files changed (1) hide show
  1. gen.py +1 -1
gen.py CHANGED
@@ -7,7 +7,7 @@ from pathlib import Path
7
 
8
  import torch
9
  from fairscale.nn.model_parallel.initialize import initialize_model_parallel
10
- from app.llama import ModelArgs, Transformer, Tokenizer, LLaMA
11
 
12
  from google.cloud import storage
13
 
 
7
 
8
  import torch
9
  from fairscale.nn.model_parallel.initialize import initialize_model_parallel
10
+ from llama import ModelArgs, Transformer, Tokenizer, LLaMA
11
 
12
  from google.cloud import storage
13