devanshty commited on
Commit
0096a37
·
verified ·
1 Parent(s): 0872985

Add model card

Browse files
Files changed (1) hide show
  1. README.md +67 -0
README.md ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - stable-diffusion
5
+ - lora
6
+ - safetensors
7
+ - text-to-image
8
+ - healthcare
9
+ - kohya-ss
10
+ base_model: runwayml/stable-diffusion-v1-5
11
+ ---
12
+
13
+ # Helios OrthoJoint
14
+
15
+ ## Model Description
16
+ Helios-OrthoJoint is a Stable Diffusion 1.5 LoRA fine-tuned for generating high-quality orthopedic and medical joint imagery. Designed for healthcare AI applications including medical illustration, patient education, and synthetic data generation for joint anatomy.
17
+
18
+ ## Model Architecture
19
+ - **Base Model**: Stable Diffusion v1.5 (`runwayml/stable-diffusion-v1-5`)
20
+ - **Fine-tuning Method**: LoRA (Low-Rank Adaptation)
21
+ - **Framework**: Diffusers / Kohya-SS
22
+ - **Task**: Text-to-Image Generation (Medical Domain)
23
+
24
+ ## Training Details
25
+ - **Training Tool**: Kohya-SS LoRA trainer
26
+ - **Domain**: Orthopedic / Joint medical imagery
27
+ - **Trigger Words**: Use medical/orthopedic descriptors in prompts for best results
28
+ - **Steps**: Fine-tuned to convergence on curated orthopedic image dataset
29
+
30
+ ## Files
31
+ | File | Description |
32
+ |------|-------------|
33
+ | `Helios_OrthoJoint_v1.safetensors` | LoRA weights (safetensors format) |
34
+
35
+ ## Usage
36
+
37
+ ```python
38
+ from diffusers import StableDiffusionPipeline
39
+ import torch
40
+ from huggingface_hub import hf_hub_download
41
+
42
+ # Download LoRA weights
43
+ lora_path = hf_hub_download(repo_id='devanshty/Helios-OrthoJoint', filename='Helios_OrthoJoint_v1.safetensors')
44
+
45
+ # Load base pipeline
46
+ pipe = StableDiffusionPipeline.from_pretrained(
47
+ "runwayml/stable-diffusion-v1-5",
48
+ torch_dtype=torch.float16
49
+ ).to("cuda")
50
+
51
+ # Load LoRA weights
52
+ pipe.load_lora_weights(lora_path)
53
+
54
+ # Generate
55
+ image = pipe("orthopedic knee joint X-ray, medical illustration, high detail").images[0]
56
+ image.save("output.png")
57
+ ```
58
+
59
+ ## Download & Use
60
+
61
+ ```python
62
+ from huggingface_hub import hf_hub_download
63
+ lora_path = hf_hub_download(repo_id='devanshty/Helios-OrthoJoint', filename='Helios_OrthoJoint_v1.safetensors')
64
+ ```
65
+
66
+ ## Disclaimer
67
+ Generated images are for research and educational use only and should not be used for clinical diagnosis.