Skip to content

Instantly share code, notes, and snippets.

View ShadowPower's full-sized avatar

夜光喵喵 ShadowPower

View GitHub Profile
@ShadowPower
ShadowPower / ds_store_cleaner.py
Created March 2, 2025 17:19
remove .DS_Store and ._*
import os
from collections import deque
# 可自定义的垃圾文件规则配置
RULES = (
{
"name": "DS_Store files",
"match": lambda n: n == ".DS_Store",
"condition": lambda e: e.stat().st_size <= 1024**2, # 1MB
},
@ShadowPower
ShadowPower / pixiv_to_webp.ps1
Last active February 22, 2025 16:37
convert pixiv ugoira to webp
param (
[Parameter(Mandatory = $true)]
[string]$PixivZipPath
)
# 检查 ffmpeg 是否可用
function Check-FFmpeg {
$ffmpegPath = Get-Command ffmpeg -ErrorAction SilentlyContinue
if (-not $ffmpegPath) {
Write-Error "ffmpeg is not installed or not in PATH."
# 安装依赖:
# pip install opencv-contrib-python
import numpy as np
import cv2
import os
from cv2.ximgproc import guidedFilter
input_dir = input('请输入存放输入图像的目录:')
@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
@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 / 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 / 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 / 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 / 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 / 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):