Joachim/VideoStory
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Actualiser genXL.py

Browse source
This commit is contained in:
Joachim Rey 2025-04-20 12:40:43 +02:00
parent 9d78d7d113
commit 6c66ecacd5
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
genXL.py
View file

@ -8,7 +8,7 @@ import os
def loadllama(): def loadllama():
llm = Llama( llm = Llama(
model_path="models/llama3.gguf", model_path="models/.gguf",
n_ctx=4096, n_ctx=4096,
n_threads=6, n_threads=6,
n_gpu_layers=-1, n_gpu_layers=-1,
@ -19,7 +19,7 @@ def loadllama():
def loadtts(): def loadtts():
model_config = outetts.GGUFModelConfig_v1( model_config = outetts.GGUFModelConfig_v1(
model_path="models/tts.gguf", model_path="models/.gguf",
language="en", language="en",
n_gpu_layers=-1, n_gpu_layers=-1,
verbose=False, verbose=False,
@ -30,7 +30,7 @@ def loadtts():
return interface, speaker return interface, speaker
def loadsdxl(): def loadsdxl():
pipe = StableDiffusionXLPipeline.from_single_file("models/sdxlReal.safetensors", torch_dtype=torch.float16, variant="fp16", requires_safety_checker=True) pipe = StableDiffusionXLPipeline.from_single_file("models/.safetensors", torch_dtype=torch.float16, variant="fp16", requires_safety_checker=True)
pipe.to("cuda") pipe.to("cuda")
pipe.enable_xformers_memory_efficient_attention() pipe.enable_xformers_memory_efficient_attention()
pipe.enable_model_cpu_offload() pipe.enable_model_cpu_offload()