Spaces:

Norod78
/

FantasyChildrenScifi-hebrew-gpt_neo-small

Running

Norod78 commited on Jun 12, 2023

Commit

5312d0a

1 Parent(s): 14a49a8

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ device = torch.device("cuda" if CUDA_AVAILABLE else "cpu")
 generator = pipeline('text-generation', model=model_id,
                      tokenizer=model_id,
-                     load_in_8bit=True,
                      device=device)
 early_stop_pattern = "\n\n\n"

 generator = pipeline('text-generation', model=model_id,
                      tokenizer=model_id,
+                     #load_in_8bit=True,
                      device=device)
 early_stop_pattern = "\n\n\n"