Model pushed

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

models/lora/pytorch_lora_weights.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5d0d20f48219b393264e82870b4e86ec51f8fc4f9d100bd572af7061449eb216
+size 787239704

requirements.txt

@@ -1 +1,2 @@
 # Specify any extra options here, like --find-links, --pre, etc. Avoid specifying dependencies here and specify them in pyproject.toml instead
+peft

src/edge_maxxing_4090_newdream.egg-info/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.1
+Name: edge-maxxing-4090-newdream
+Version: 1.0.0
+Summary: An edge-maxxing model submission for the 4090 newdream contest
+Requires-Python: <3.11,>=3.10
+Requires-Dist: diffusers==0.30.2
+Requires-Dist: transformers==4.41.2
+Requires-Dist: accelerate==0.31.0
+Requires-Dist: omegaconf==2.3.0
+Requires-Dist: edge-maxxing-pipelines@ git+https://github.com/womboai/edge-maxxing#subdirectory=pipelines

src/edge_maxxing_4090_newdream.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+README.md
+pyproject.toml
+src/main.py
+src/pipeline.py
+src/edge_maxxing_4090_newdream.egg-info/PKG-INFO
+src/edge_maxxing_4090_newdream.egg-info/SOURCES.txt
+src/edge_maxxing_4090_newdream.egg-info/dependency_links.txt
+src/edge_maxxing_4090_newdream.egg-info/entry_points.txt
+src/edge_maxxing_4090_newdream.egg-info/requires.txt
+src/edge_maxxing_4090_newdream.egg-info/top_level.txt

src/edge_maxxing_4090_newdream.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

src/edge_maxxing_4090_newdream.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+start_inference = main:main

src/edge_maxxing_4090_newdream.egg-info/requires.txt

@@ -0,0 +1,5 @@
+diffusers==0.30.2
+transformers==4.41.2
+accelerate==0.31.0
+omegaconf==2.3.0
+edge-maxxing-pipelines@ git+https://github.com/womboai/edge-maxxing#subdirectory=pipelines

src/edge_maxxing_4090_newdream.egg-info/top_level.txt

@@ -0,0 +1,2 @@
+main
+pipeline

src/pipeline.py

@@ -1,16 +1,40 @@
 import torch
 from PIL.Image import Image
-from diffusers import StableDiffusionXLPipeline
+from diffusers import StableDiffusionXLPipeline, LCMScheduler
 from pipelines.models import TextToImageRequest
 from torch import Generator
 
 
 def load_pipeline() -> StableDiffusionXLPipeline:
- pipeline = StableDiffusionXLPipeline.from_pretrained(
+ pipeline: StableDiffusionXLPipeline = StableDiffusionXLPipeline.from_pretrained(
  "./models/newdream-sdxl-20",
  torch_dtype=torch.float16,
  local_files_only=True,
  ).to("cuda")
+ adapter_id = "./models/lora"
+ pipeline.scheduler = LCMScheduler.from_config({
+ "_class_name": "LCMScheduler",
+ "_diffusers_version": "0.29.0",
+ "beta_end": 0.012,
+ "beta_schedule": "scaled_linear",
+ "beta_start": 0.00085,
+ "clip_sample": False,
+ "clip_sample_range": 1.0,
+ "num_train_timesteps": 1000,
+ "original_inference_steps": 300,
+ "prediction_type": "epsilon",
+ "rescale_betas_zero_snr": False,
+ "sample_max_value": 1.0,
+ "set_alpha_to_one": False,
+ "steps_offset": 1,
+ "thresholding": False,
+ "timestep_scaling": 10.0,
+ "timestep_spacing": "leading",
+ "trained_betas": None,
+ })
+
+ # # LCMScheduler.from_config(pipeline.scheduler.config)
+ pipeline.load_lora_weights(adapter_id)
 
  pipeline(prompt="")
 
@@ -26,4 +50,6 @@ def infer(request: TextToImageRequest, pipeline: StableDiffusionXLPipeline) -> I
  width=request.width,
  height=request.height,
  generator=generator,
+ num_inference_steps=4,
+ guidance_scale=1.5,
  ).images[0]