lxe commited on
Commit
d3bfed8
1 Parent(s): bc97c87

Added duplication instructions

Browse files
Files changed (1) hide show
  1. main.py +7 -1
main.py CHANGED
@@ -433,9 +433,15 @@ def inference_tab():
433
  with gr.Blocks(
434
  css="#refresh-button { max-width: 32px }",
435
  title="Simple LLaMA Finetuner") as demo:
436
- gr.HTML("<div style=\"display: flex; align-items: center; justify-content: center;\"><h3 style=\"margin: 0\"><a target=\"_blank\" href=https://github.com/lxe/simple-llama-finetuner>Simple LLaMA Finetuner</a></h3></div>")
 
 
 
437
  training_tab()
438
  inference_tab()
 
 
 
439
 
440
  if __name__ == '__main__':
441
  parser = argparse.ArgumentParser(description="Simple LLaMA Finetuner")
 
433
  with gr.Blocks(
434
  css="#refresh-button { max-width: 32px }",
435
  title="Simple LLaMA Finetuner") as demo:
436
+ gr.Markdown("""
437
+ ## 🦙 Simple LLaMA Finetuner [<img src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&amp;style=flat&amp;logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&amp;logoWidth=14" alt="" style="display: inline;">](https://huggingface.co/spaces/lxe/simple-llama-finetuner?duplicate=true)
438
+ This tunes the [llama-7b](https://huggingface.co/decapoda-research/llama-7b-hf) model on your own text. Duplicate this space onto a GPU-enabled space to run.
439
+ """)
440
  training_tab()
441
  inference_tab()
442
+ gr.Markdown("""
443
+ Enter your samples separated by two blank lines, then click "Train" to start training a new LoRA model. Once the model is trained, you can use it to generate new text by entering a prompt and clicking "Generate".
444
+ """)
445
 
446
  if __name__ == '__main__':
447
  parser = argparse.ArgumentParser(description="Simple LLaMA Finetuner")