Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ from share_btn import community_icon_html, loading_icon_html, share_js, share_bt
|
|
| 12 |
|
| 13 |
checkpoint = "smallcloudai/Refact-1_6B-fim"
|
| 14 |
device = "cuda"
|
| 15 |
-
device = "cpu" # for GPU usage or "cpu" for CPU usage
|
| 16 |
|
| 17 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
| 18 |
model = AutoModelForCausalLM.from_pretrained(checkpoint, trust_remote_code=True).to(device)
|
|
@@ -116,7 +116,7 @@ description = """
|
|
| 116 |
<ul>
|
| 117 |
<li><a href="https://huggingface.co/smallcloudai/Refact-1_6B-fim" style='color: #e6b800;'>ReFact 1.6B</a>: An Open-Source Coding Assistant with Fine-Tuning on codebase, autocompletion, code refactoring, code analysis, integrated chat and more</li>
|
| 118 |
</ul>
|
| 119 |
-
<p><b>Please note:</b> This demo is based on the Big Code Playground. It is running on
|
| 120 |
</div>
|
| 121 |
"""
|
| 122 |
|
|
|
|
| 12 |
|
| 13 |
checkpoint = "smallcloudai/Refact-1_6B-fim"
|
| 14 |
device = "cuda"
|
| 15 |
+
#device = "cpu" # for GPU usage or "cpu" for CPU usage
|
| 16 |
|
| 17 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
| 18 |
model = AutoModelForCausalLM.from_pretrained(checkpoint, trust_remote_code=True).to(device)
|
|
|
|
| 116 |
<ul>
|
| 117 |
<li><a href="https://huggingface.co/smallcloudai/Refact-1_6B-fim" style='color: #e6b800;'>ReFact 1.6B</a>: An Open-Source Coding Assistant with Fine-Tuning on codebase, autocompletion, code refactoring, code analysis, integrated chat and more</li>
|
| 118 |
</ul>
|
| 119 |
+
<p><b>Please note:</b> This demo is based on the Big Code Playground. It is running on GPU, but can also be run on CPU.</p>
|
| 120 |
</div>
|
| 121 |
"""
|
| 122 |
|