Spaces:

anton-l
/

rudall-e

Runtime error

anton-l HF Staff commited on Dec 16, 2021

Commit

5e05aea

1 Parent(s): 91602b2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ tqdm.__init__ = partialmethod(tqdm.__init__, disable=True)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 translation_model = FSMTForConditionalGeneration.from_pretrained("facebook/wmt19-en-ru", torch_dtype=torch.float16).to(device)
-tokenizer = FSMTTokenizer.from_pretrained("facebook/wmt19-en-ru")
 dalle = get_rudalle_model("Malevich", pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()
 vae = get_vae().to(device)

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 translation_model = FSMTForConditionalGeneration.from_pretrained("facebook/wmt19-en-ru", torch_dtype=torch.float16).to(device)
+translation_tokenizer = FSMTTokenizer.from_pretrained("facebook/wmt19-en-ru")
 dalle = get_rudalle_model("Malevich", pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()
 vae = get_vae().to(device)