Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -99,7 +99,11 @@ def load_new_model(model_id):
|
|
| 99 |
except Exception as e:
|
| 100 |
yield f"Error loading model: {str(e)}"
|
| 101 |
|
|
|
|
| 102 |
@spaces.GPU
|
|
|
|
|
|
|
|
|
|
| 103 |
def run_inference(user_prompt, max_tokens, temperature, top_k, top_p, rep_penalty, ngram_size, do_sample):
|
| 104 |
"""Generates text via streaming generator."""
|
| 105 |
if model_manager.model is None or model_manager.tokenizer is None:
|
|
|
|
| 99 |
except Exception as e:
|
| 100 |
yield f"Error loading model: {str(e)}"
|
| 101 |
|
| 102 |
+
|
| 103 |
@spaces.GPU
|
| 104 |
+
def dummy():
|
| 105 |
+
pass
|
| 106 |
+
|
| 107 |
def run_inference(user_prompt, max_tokens, temperature, top_k, top_p, rep_penalty, ngram_size, do_sample):
|
| 108 |
"""Generates text via streaming generator."""
|
| 109 |
if model_manager.model is None or model_manager.tokenizer is None:
|