Spaces:

Norod78
/

FantasyChildrenScifi-hebrew-gpt_neo-small

Running

Norod78 commited on Jun 12, 2023

Commit

2eb7ad1

•

1 Parent(s): 5312d0a

Upload 3 files

Files changed (2) hide show

README.md CHANGED Viewed

@@ -4,7 +4,7 @@ emoji: 🧙‍♀️
 colorFrom: yellow
 colorTo: blue
 sdk: gradio
-sdk_version: 3.24.1
 app_file: app.py
 pinned: false
 license: mit

 colorFrom: yellow
 colorTo: blue
 sdk: gradio
+sdk_version: 3.28.1
 app_file: app.py
 pinned: false
 license: mit

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"