How to use hchen657Naza/PWM with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("hchen657Naza/PWM", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0]
https://huggingface.co/hchen657Naza/PWM/resolve/main/PWM_code/policy_rl/__init__.py
hf download hf://hchen657Naza/PWM/PWM_code/policy_rl/__init__.py
curl -L -o __init__.py https://huggingface.co/hchen657Naza/PWM/resolve/main/PWM_code/policy_rl/__init__.py
