Skip to content

Instantly share code, notes, and snippets.

View ShadowPower's full-sized avatar

夜光喵喵 ShadowPower

View GitHub Profile
@ShadowPower
ShadowPower / ShellArgs.java
Last active May 27, 2019 00:37
有人说想要个拆分Shell命令和参数的工具,转义只需要支持\" \'
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
enum State {
NORMAL,SPACE,SINGLE,DOUBLE,ESCAPE
}
class Extractor {
private State state; // 当前状态
@ShadowPower
ShadowPower / mspaint.qml
Last active October 11, 2019 16:35
QML doodle canvas
import QtQuick 2.12
import QtQuick.Window 2.12
Window {
id: msPaint
visible: true
width: 640
height: 480
title: qsTr("MoeSoftPaint")
@ShadowPower
ShadowPower / clip-multilingual-onnx.py
Last active October 28, 2022 02:38
convert sentence transformers multilingual clip model to onnx
from os import makedirs
from os.path import dirname, abspath
from pathlib import Path
from onnxruntime.quantization import quantize_dynamic, QuantType
from sentence_transformers import SentenceTransformer, util
import torch
from torch import nn
from transformers.onnx.convert import export_pytorch
@ShadowPower
ShadowPower / sd_onnx.py
Created November 13, 2022 13:26
stable diffusion onnx exporter for bes-dev
# Use https://github.com/harishanand95/diffusers to export
from pathlib import Path
import torch
import torch.nn as nn
from diffusers import StableDiffusionPipeline, LMSDiscreteScheduler, DiffusionPipeline
from transformers import CLIPTextModel
class VQEncoder(nn.Module):
@ShadowPower
ShadowPower / letterbox.py
Created February 21, 2023 08:15
用新版bing生成的调整图片尺寸&加上白色边框
import os
from PIL import Image
# 定义源目录和目标目录
source_dir = r"D:\dataset\before"
target_dir = r"D:\dataset\after"
# 遍历源目录下的所有图片文件
for filename in os.listdir(source_dir):
# 打开图片文件
@ShadowPower
ShadowPower / player.cpp
Created July 5, 2023 03:02
ffmpeg playback with c++
#include "player.h"
using namespace std;
void data_callback(ma_device* pDevice, void* pOutput, const void* pInput, ma_uint32 frameCount) {
AVAudioFifo* fifo = reinterpret_cast<AVAudioFifo*>(pDevice->pUserData);
av_audio_fifo_read(fifo, &pOutput, frameCount);
(void) pInput;
}
@ShadowPower
ShadowPower / Cargo.toml
Created July 5, 2023 03:04
ffmpeg playback with rust
[package]
name = "player"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
cpal = "0.13.5"
ffmpeg-next = "5.0.3"
@ShadowPower
ShadowPower / baichuan.py
Last active August 2, 2023 03:11
baichuan-13b quantized
import json
import torch
import streamlit as st
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
from transformers.generation.utils import GenerationConfig
st.set_page_config(page_title="Baichuan-13B-Chat")
st.title("Baichuan-13B-Chat")
@ShadowPower
ShadowPower / baichuan-13b.ipynb
Created July 14, 2023 07:04
Baichuan 13B for Google Colab
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@ShadowPower
ShadowPower / qwen_web.py
Created August 3, 2023 15:15
通义千问 7B Gradio
import gradio as gr
import mdtex2html
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
from transformers.generation.utils import GenerationConfig
# 模型路径
MODEL_PATH = 'Qwen/Qwen-7B-Chat'
CONTEXT_SIZE = 8192