Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-21 23:57:58 +01:00
Code Issues Packages Projects Releases Wiki Activity

Reorder the imports here too

Browse Source
This commit is contained in:
oobabooga 2023-02-10 15:57:55 -03:00
parent 219366342b
commit 76d3d7ddb3
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
convert-to-torch.py
View File

@ -6,11 +6,12 @@ python convert-to-torch.py models/opt-1.3b
The output will be written to torch-dumps/name-of-the-model.pt
'''
from transformers import AutoModelForCausalLM
import torch
from sys import argv
from pathlib import Path
from sys import argv
import torch
from transformers import AutoModelForCausalLM
path = Path(argv[1])
model_name = path.name

2
modules/html_generator.py
View File

@ -4,9 +4,9 @@ This is a library for formatting GPT-4chan and chat outputs as nice HTML.
'''
import copy
import re
from pathlib import Path
import copy
def generate_basic_html(s):
s = '\n'.join([f'<p style="margin-bottom: 20px">{line}</p>' for line in s.split('\n')])