Spaces:

Tarive
/

HRM

Build error

HRM / Dockerfile

Saksham Adhikari

Add Dockerfile and startup script for HF Space

bb38358 about 1 year ago

407 Bytes

# Start from a PyTorch + CUDA base image
FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-devel

WORKDIR /code
COPY . .

# Install dependencies
RUN pip install -r requirements.txt
RUN pip install flash-attn --no-build-isolation

# Make the startup script executable
COPY startup.sh .
RUN chmod +x startup.sh

# This is the new command: it runs your script automatically when the Space starts.
CMD ["./startup.sh"]

	# Start from a PyTorch + CUDA base image
	FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-devel

	WORKDIR /code
	COPY . .

	# Install dependencies
	RUN pip install -r requirements.txt
	RUN pip install flash-attn --no-build-isolation

	# Make the startup script executable
	COPY startup.sh .
	RUN chmod +x startup.sh

	# This is the new command: it runs your script automatically when the Space starts.
	CMD ["./startup.sh"]