Update app.py
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ def get_model():
|
|
41 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln61Paraphrase")
|
42 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln66Paraphrase")
|
43 |
tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln65Paraphrase")
|
44 |
-
model =
|
45 |
return model, tokenizer
|
46 |
|
47 |
model, tokenizer = get_model()
|
|
|
41 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln61Paraphrase")
|
42 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln66Paraphrase")
|
43 |
tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln65Paraphrase")
|
44 |
+
model = GPTJForCausalLM.from_pretrained("EleutherAI/gpt-j-6B", revision="float16", torch_dtype=torch.float16, low_cpu_mem_usage=True)
|
45 |
return model, tokenizer
|
46 |
|
47 |
model, tokenizer = get_model()
|