rajistics commited on
Commit
310157b
1 Parent(s): 68b6ff1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 CPU, but can be run faster using GPU.</p>
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