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 |
|